Update app.py

app.py

@@ -1,73 +1,108 @@
-import gradio as gr
+import gradio as gr 
 from huggingface_hub import InferenceClient
 
 # ✅ Connect to Hugging Face API (Replace with your model or API key)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# ✅ Language Options
+titles = {
+    "en": {
+        "header": "🧠 Bandar AI - Your Smart Assistant",
+        "subtitle": "💬 Talk to me! I'm here to help you with anything you need.",
+        "chat_label": "Bandar AI",
+        "input_label": "Type your message here",
+        "placeholder": "Hello! How can I help you?",
+        "send_button": "Send",
+        "clear_button": "Clear Chat",
+        "toggle_button": "Switch to Arabic"
+    },
+    "ar": {
+        "header": "🧠 بندر AI - مساعدك الذكي",
+        "subtitle": "💬 تحدث معي! أنا هنا لمساعدتك في أي موضوع تحتاجه.",
+        "chat_label": "بندر AI",
+        "input_label": "اكتب رسالتك هنا",
+        "placeholder": "مرحبًا! كيف يمكنني مساعدتك؟",
+        "send_button": "إرسال",
+        "clear_button": "مسح المحادثة",
+        "toggle_button": "التبديل إلى الإنجليزية"
+    }
+}
+
 # ✅ System Prompt for Bandar AI
 BANDAR_AI_PROMPT = (
-    "أنت مساعد ذكاء اصطناعي يدعى بندر AI. "
+    "أنت مساعد ذكاء اصطناعي يدعى باندر AI. "
     "تهدف إلى تقديم المساعدة في مجموعة متنوعة من المواضيع، مثل الإجابة على الأسئلة، تقديم النصائح، والمساعدة في المهام اليومية. "
     "كن ودودًا ومتعاونًا دائمًا."
 )
 
-# ✅ Initialize Conversation History (Stateful Interaction)
+# ✅ Initialize Conversation History
 def initialize_history():
     return [{"role": "system", "content": BANDAR_AI_PROMPT}]
 
 # ✅ Chat Response Function
 def respond(message, history):
-    # Add the user's message to the conversation history
     history.append({"role": "user", "content": message})
     
-    # Generate a response from the model
     response = ""
     for msg in client.chat_completion(history, max_tokens=512, stream=True, temperature=0.7, top_p=0.95):
         token = msg.choices[0].delta.content
         if token:
             response += token
-            yield history + [{"role": "assistant", "content": response}]  # Stream the response
+            yield [(msg["content"], history[i + 1]["content"]) for i, msg in enumerate(history[:-1]) if msg["role"] == "user"] + [(message, response)]
     
-    # Add the assistant's response to the conversation history
     history.append({"role": "assistant", "content": response})
 
 # ✅ Clear Conversation History
 def clear_history():
-    return initialize_history(), []  # Reset history and chatbot display
+    return initialize_history(), []
+
+# ✅ Toggle Language Function
+def toggle_language(current_language):
+    return "ar" if current_language == "en" else "en"
 
 # ✅ Customized UI with Gradio Components
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="teal")) as demo:
-    gr.Markdown(
-        """
-        # 🧠 **Bandar AI - Your Smart Assistant**  
-        ### 💬 تحدث معي! أنا هنا لمساعدتك في أي موضوع تحتاجه.
-        """
-    )
+    # Language state
+    language = gr.State("en")  # Default to English
     
-    # State to maintain conversation history
+    # UI Elements
+    header = gr.Markdown(titles["en"]["header"])
+    subtitle = gr.Markdown(titles["en"]["subtitle"])
     history = gr.State(initialize_history())
-    
-    # Chatbot Component for Displaying Conversations
-    chatbot = gr.Chatbot(label="Bandar AI", bubble_full_width=False)
-    
-    # Textbox for User Input
+    chatbot = gr.Chatbot(label=titles["en"]["chat_label"], bubble_full_width=False)
+
     with gr.Row():
-        user_input = gr.Textbox(
-            label="اكتب رسالتك هنا",
-            placeholder="مرحبًا! كيف يمكنني مساعدتك؟",
-            scale=8
-        )
-        send_button = gr.Button("إرسال", variant="primary", scale=2)
+        user_input = gr.Textbox(label=titles["en"]["input_label"], placeholder=titles["en"]["placeholder"], scale=8)
+        send_button = gr.Button(titles["en"]["send_button"], variant="primary", scale=2)
     
-    # Buttons for Clearing History
     with gr.Row():
-        clear_button = gr.Button("مسح المحادثة", variant="secondary")
-    
+        clear_button = gr.Button(titles["en"]["clear_button"], variant="secondary")
+        toggle_button = gr.Button(titles["en"]["toggle_button"], variant="secondary")
+
     # Event Handlers
     send_button.click(respond, inputs=[user_input, history], outputs=[chatbot, history])
     user_input.submit(respond, inputs=[user_input, history], outputs=[chatbot, history])
     clear_button.click(clear_history, outputs=[history, chatbot])
 
+    # Toggle Language
+    def update_ui(lang):
+        return (
+            titles[lang]["header"],
+            titles[lang]["subtitle"],
+            titles[lang]["chat_label"],
+            titles[lang]["input_label"],
+            titles[lang]["placeholder"],
+            titles[lang]["send_button"],
+            titles[lang]["clear_button"],
+            titles[lang]["toggle_button"]
+        )
+
+    toggle_button.click(toggle_language, inputs=[language], outputs=[language])
+    toggle_button.click(
+        update_ui, inputs=[language], 
+        outputs=[header, subtitle, chatbot, user_input, user_input, send_button, clear_button, toggle_button]
+    )
+
 # ✅ Run the Chatbot
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()