Update app.py

app.py

@@ -1,31 +1,61 @@
 import gradio as gr
 from transformers import pipeline
+import requests
 
-qa_pipeline = pipeline("question-answering", model="hosseinfarahi/squad-roberta-large")
+qa = pipeline("question-answering", model="hosseinfarahi/squad-roberta-large")
+fa2en = pipeline("translation", model="Helsinki-NLP/opus-mt-fa-en")
+en2fa = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fa")
 
-context = ""
+CONTEXTS = {
+    "🍝 پاستا": "To cook pasta, boil water, add salt and oil. Cook dry pasta for 8 to 10 minutes, then drain and serve.",
+    "💰 بیت‌کوین": "Bitcoin is a decentralized digital currency launched in 2009 by a mysterious person or group called Satoshi Nakamoto.",
+    "🐍 پایتون": "Python is a versatile programming language often used for web development, data science, AI, and scripting.",
+    "🧠 پزشکی": "Medical advice should always be validated by a doctor. Common cold symptoms include sore throat, runny nose, and fever.",
+    "🥗 تغذیه": "A healthy diet includes vegetables, fruits, whole grains, and lean proteins. Drinking water is essential.",
+    "🇬🇧 زبان انگلیسی": "English grammar includes verb tenses, articles, and sentence structure. Practice improves fluency."
+}
 
-def chat_fn(message, history):
-    global context
-    if message.lower().startswith("context:") or message.startswith("متن:"):
-        context = message.split(":", 1)[1].strip()
-        return "متن جدید ثبت شد. حالا سوالتو بپرس."
+selected_topic = "🍝 پاستا"
 
-    if not context:
-        return "لطفاً ابتدا یک متن وارد کن. مثلا: context: فلان پاراگراف..."
+def set_topic(choice):
+    global selected_topic
+    selected_topic = choice
+    return f"✅ موضوع انتخاب‌شده: {choice}"
 
-    try:
-        result = qa_pipeline(question=message, context=context)
-        return result["answer"]
-    except Exception as e:
-        return f"خطا در پاسخ‌دهی: {str(e)}"
+def smart_chat(message, history):
+    context = CONTEXTS[selected_topic]
+    if "bitcoin" in selected_topic.lower():
+        try:
+            res = requests.get("https://api.coingecko.com/api/v3/simple/price?ids=bitcoin&vs_currencies=usd")
+            price = res.json()["bitcoin"]["usd"]
+            context += f"\nCurrent Bitcoin price is ${price}"
+        except:
+            context += "\n(خطا در دریافت قیمت لحظه‌ای بیت‌کوین)"
 
-chatbot = gr.ChatInterface(
-    fn=chat_fn,
-    title="🤖 چت‌بات هوشمند پرسش و پاسخ",
-    description="اول یک متن بده (مثلاً: context: پاستا را ۱۰ دقیقه بجوشانید...)، بعد سوال بپرس.",
-    examples=["context: Python is a programming language...", "What is Python?", "How long to cook pasta?"]
+    question_en = fa2en(message)[0]['translation_text']
+    result = qa(question=question_en, context=context)
+    answer_en = result["answer"]
+    answer_fa = en2fa(answer_en)[0]['translation_text']
+    return answer_fa
+
+topic_selector = gr.Dropdown(
+    label="📚 انتخاب موضوع",
+    choices=list(CONTEXTS.keys()),
+    value="🍝 پاستا",
+    interactive=True,
+    info="یکی از موضوعات را انتخاب کن",
+    type="value"
+)
+
+chat = gr.ChatInterface(
+    fn=smart_chat,
+    chatbot=gr.Chatbot(show_label=False),
+    textbox=gr.Textbox(placeholder="سوالت رو بنویس و اینتر بزن...", container=False),
+    title="🤖 چت‌بات فارسی چندمنظوره",
+    description="✅ اول از لیست بالا موضوع انتخاب کن بعد شروع کن به سوال پرسیدن...",
+    additional_inputs=[topic_selector],
+    additional_inputs_placement="sidebar"
 )
 
 if __name__ == "__main__":
-    chatbot.launch()
+    chat.launch()