Update ui.py

ui.py

@@ -1,20 +1,15 @@
 import gradio as gr
 import requests
 
-API_URL = "http://localhost:8000/chat/"  # تغییر به URL واقعی بک‌اند در آینده
+def chat_with_ai(input_text, mode):
+    response = requests.post("http://localhost:8000/chat", json={"input_text": input_text, "mode": mode})
+    return response.json()["response"]
 
-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",
+ui = gr.Interface(
+    fn=chat_with_ai,
+    inputs=["text", gr.Radio(["general", "iran", "math"], label="Mode")],
     outputs="text",
-    title="ChatGPT Clone",
-    description="چت‌بات بر پایه GPT-Neo"
+    title="AI Chat System",
 )
 
-if __name__ == "__main__":
-    interface.launch()
+ui.launch()