Create ui.py

ui.py

@@ -0,0 +1,20 @@
+import gradio as gr
+import requests
+
+API_URL = "http://localhost:8000/chat/"  # تغییر به URL واقعی بک‌اند در آینده
+
+def ask_chatbot(query):
+    response = requests.post(API_URL, json={"query": query})
+    return response.json().get("response", "خطایی رخ داد!")
+
+# رابط کاربری
+interface = gr.Interface(
+    fn=ask_chatbot,
+    inputs="text",
+    outputs="text",
+    title="ChatGPT Clone",
+    description="چت‌بات بر پایه GPT-Neo"
+)
+
+if __name__ == "__main__":
+    interface.launch()