Update app.py

app.py

@@ -4,44 +4,70 @@ from huggingface_hub import InferenceClient
 # ✅ Connect to Hugging Face API (Replace with your model or API key)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# ✅ System Prompt for Bandar AI (General Purpose)
+# ✅ System Prompt for Bandar AI
 BANDAR_AI_PROMPT = (
-    "أنت مساعد ذكاء اصطناعي يدعى باندر AI. "
+    "أنت مساعد ذكاء اصطناعي يدعى بندر AI. "
     "تهدف إلى تقديم المساعدة في مجموعة متنوعة من المواضيع، مثل الإجابة على الأسئلة، تقديم النصائح، والمساعدة في المهام اليومية. "
     "كن ودودًا ومتعاونًا دائمًا."
 )
 
 # ✅ Initialize Conversation History (Stateful Interaction)
-conversation_history = [{"role": "system", "content": BANDAR_AI_PROMPT}]
+def initialize_history():
+    return [{"role": "system", "content": BANDAR_AI_PROMPT}]
 
 # ✅ Chat Response Function
-def respond(message):
-    global conversation_history  # Use global variable to maintain state
-    
+def respond(message, history):
     # Add the user's message to the conversation history
-    conversation_history.append({"role": "user", "content": message})
+    history.append({"role": "user", "content": message})
     
     # Generate a response from the model
     response = ""
-    for msg in client.chat_completion(conversation_history, max_tokens=512, stream=True, temperature=0.7, top_p=0.95):
+    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 response  # Stream the response
+            yield history + [{"role": "assistant", "content": response}]  # Stream the response
     
     # Add the assistant's response to the conversation history
-    conversation_history.append({"role": "assistant", "content": response})
+    history.append({"role": "assistant", "content": response})
+
+# ✅ Clear Conversation History
+def clear_history():
+    return initialize_history(), []  # Reset history and chatbot display
 
-# ✅ Clean & Modern UI
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+# ✅ 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 Stateful Assistant**  
-        ### 💬 تحدث معي! أنا أتذكر ما قلته سابقًا لجعل المحادثة أكثر طبيعية.
+        # 🧠 **Bandar AI - Your Smart Assistant**  
+        ### 💬 تحدث معي! أنا هنا لمساعدتك في أي موضوع تحتاجه.
         """
     )
-    chatbot = gr.ChatInterface(respond)
+    
+    # State to maintain conversation history
+    history = gr.State(initialize_history())
+    
+    # Chatbot Component for Displaying Conversations
+    chatbot = gr.Chatbot(label="Bandar AI", bubble_full_width=False)
+    
+    # Textbox for User Input
+    with gr.Row():
+        user_input = gr.Textbox(
+            label="اكتب رسالتك هنا",
+            placeholder="مرحبًا! كيف يمكنني مساعدتك؟",
+            scale=8
+        )
+        send_button = gr.Button("إرسال", variant="primary", scale=2)
+    
+    # Buttons for Clearing History
+    with gr.Row():
+        clear_button = gr.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])
 
-# ✅ Run the chatbot
+# ✅ Run the Chatbot
 if __name__ == "__main__":
     demo.launch()