Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 import requests
 import json
 import os
@@ -16,17 +16,22 @@ def generate_response(prompt):
     if response.status_code == 200:
         return response.json()
     else:
-        st.error(f"Error: {response.status_code}")
-        st.error(response.text)
-        return None
+        return f"Error: {response.status_code}\n{response.text}"
 
-def main():
-    st.title("Code Assistant")
-    prompt = st.text_area("Enter your prompt", height=100)
-    if st.button("Submit"):
-        response = generate_response(prompt)
-        if response:
-            st.text_area("Response", value=json.dumps(response, indent=2), height=200)
+def main(prompt):
+    response = generate_response(prompt)
+    if isinstance(response, str):
+        return response
+    else:
+        return json.dumps(response, indent=2)
+
+iface = gr.Interface(
+    fn=main,
+    inputs=gr.Textbox(lines=4, placeholder="Enter your prompt here..."),
+    outputs=gr.Textbox(lines=10),
+    title="Code Assistant",
+    description="Enter your prompt and get responses from the model."
+)
 
 if __name__ == "__main__":
-    main()
+    iface.launch()