Update app.py

app.py

@@ -1,58 +1,45 @@
 import os
 import gradio as gr
 from ollama import Client
+import time
 
-# 🧩 دریافت API Key از Hugging Face Secrets
 OLLAMA_API_KEY = os.environ.get("OLLAMA_API_KEY")
-
 if not OLLAMA_API_KEY:
-    raise ValueError("API Key تنظیم نشده است. لطفاً در بخش Secrets مقدار OLLAMA_API_KEY را وارد کنید.")
+    raise ValueError("API Key تنظیم نشده است.")
 
-# 🧠 اتصال به Ollama Cloud
 client = Client(
     host="https://ollama.com",
     headers={"Authorization": f"Bearer {OLLAMA_API_KEY}"}
 )
 
-# مدل‌های Cloud در دسترس
 AVAILABLE_MODELS = [
     "gpt-oss:20b-cloud",
-    "gpt-oss:120b-cloud",
-    "qwen3-coder:480b-cloud",
-    "deepseek-v3.1:671b-cloud",
-    "kimi-k2:1t-cloud"
+    "qwen3-coder:480b-cloud"
 ]
 
-# تابع چت با استریمینگ
-def chat_fn(message, history, model_name):
+# تابع استریمینگ
+def respond(message, history, model_name):
     messages = []
-    # تبدیل تاریخچه چت
     for h in history:
         messages.append({"role": "user", "content": h[0]})
         messages.append({"role": "assistant", "content": h[1]})
     messages.append({"role": "user", "content": message})
 
-    response = ""
-    # استریم پاسخ مدل
+    bot_message = ""
+    yield history, ""  # برای ریست کردن Textbox کاربر
     for part in client.chat(model_name, messages=messages, stream=True):
-        chunk = part["message"]["content"]
-        response += chunk
-        yield response  # ارسال زنده به Gradio
-
-# 🖥 رابط کاربری Gradio
-with gr.Blocks(title="💬 Ollama Cloud Chatbot") as demo:
-    gr.Markdown("# 💬 Ollama Cloud Chatbot\nچت با مدل‌های ابری Ollama بدون GPU 🚀")
-
-    with gr.Row():
-        model_selector = gr.Dropdown(
-            choices=AVAILABLE_MODELS,
-            value="gpt-oss:20b-cloud",
-            label="انتخاب مدل"
-        )
-
-    chatbot = gr.ChatInterface(
-        fn=lambda msg, hist: chat_fn(msg, hist, model_selector.value),
-        title="Ollama Cloud Chatbot"
-    )
+        bot_message += part["message"]["content"]
+        yield history + [(message, bot_message)], ""  # بروزرسانی استریم
+        time.sleep(0.01)  # کمی تاخیر برای ظاهر شدن استریم روان
+
+# رابط Blocks
+with gr.Blocks() as demo:
+    gr.Markdown("# 💬 Ollama Cloud Chatbot (Streamed)")
+    model_selector = gr.Dropdown(choices=AVAILABLE_MODELS, value=AVAILABLE_MODELS[0], label="انتخاب مدل")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="پیام خود را وارد کنید")
+    send_btn = gr.Button("ارسال")
+
+    send_btn.click(respond, inputs=[msg, chatbot, model_selector], outputs=[chatbot, msg])
 
 demo.launch()