Update app.py

app.py

@@ -1,6 +1,7 @@
 import google.generativeai as palm
-import streamlit as st 
+import gradio as gr
 import os 
+import json
 
 # Set your API key
 palm.configure(api_key=os.environ['PALM_KEY'])
@@ -9,10 +10,20 @@ palm.configure(api_key=os.environ['PALM_KEY'])
 # model = 'models/text-bison-001'
 
 
-if prompt := st.text_input("Write your message: "):
-    response = palm.chat(messages=prompt)
-    with st.chat_message("assistant"):
-        st.write(response.last)
-
+def responsenew(data):
+    response = palm.chat(messages=data)
+    respo = {
+        "message": response.last,
+        "action": "nothing",
+        "function": "nothing"
+    }
+    print(data)
+    return json.dumps(respo)
 
+gradio_interface = gr.Interface(
+  fn = responsenew,
+  inputs = "text",
+  outputs = "text" 
+)
+gradio_interface.launch()