Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import gradio as gr
-from streamlit_gradio import st_gradio
 
 def chat_with_model(input_text):
     # Call the chat model and return the response
@@ -25,11 +24,15 @@ def main():
         st.markdown("---")
         st.write(response)
 
-    # Create the Gradio interface
-    gr_interface = gr.Interface(fn=chat_with_model, inputs="text", outputs="text")
-    
-    # Render the Gradio interface using the st_gradio function
-    st_gradio(gr_interface)
+    # Render the Gradio interface using Streamlit's st.form and st.button
+    with st.form(key="my_form"):
+        text_input = st.text_input("Enter your question")
+        submit_button = st.form_submit_button(label="Ask")
+        if submit_button:
+            with st.spinner("Thinking..."):
+                response = chat_with_model(text_input)
+            st.markdown("---")
+            st.write(response)
 
 if __name__ == "__main__":
     main()