Update app.py

app.py

@@ -3,8 +3,7 @@ from transformers import pipeline
 import gradio as gr
 import PyPDF2
 import os
-from huggingface_hub import login, HfFolder
-from getpass import getpass
+from huggingface_hub import login
 
 # Space configuration
 SPACE_DIR = os.environ.get("HF_HOME", os.getcwd())
@@ -64,11 +63,6 @@ SYSTEM_PROMPT = f"""You are Foton, Swahili AI assistant. Tasks:
 Maintain friendly, patient demeanor with cultural context.
 """
 
-# Update the WELCOME_MESSAGE format
-WELCOME_MESSAGE = {
-    "role": "assistant",
-    "content": "**Karibu Lugha Tausi!** Mimi ni Foton, msaidizi wako wa Kiswahili. Niko hapa kukusaidia kujifunza na kuzungumza Kiswahili. **Ninaweza kukusaidiaje leo?** 😊"
-}
 def format_messages(history):
     """Format chat history with system prompt"""
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
@@ -77,13 +71,12 @@ def format_messages(history):
             messages.append(msg)
     return messages
 
-def generate_response(message, history):
+def generate_response(history):
     """Generate AI response"""
     try:
-        formatted_history = format_messages(history)
-        formatted_history.append({"role": "user", "content": message})
+        messages = format_messages(history)
         
-        prompt = "\n".join([f"{m['role']}: {m['content']}" for m in formatted_history])
+        prompt = "\n".join([f"{m['role']}: {m['content']}" for m in messages])
         
         output = pipe(
             prompt,
@@ -102,39 +95,33 @@ def generate_response(message, history):
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Lugha Tausi - Swahili Assistant")
     
-    # Update the Chatbot initialization
     chatbot = gr.Chatbot(
-    value=[WELCOME_MESSAGE],  # Now using proper message format
-    height=600,
-    show_label=False,
-    avatar_images=(None, "user.png"),
-    bubble_full_width=False,
-    show_share_button=False,
-    type="messages"
+        value=[{"role": "assistant", "content": "**Karibu Lugha Tausi!** Mimi ni Foton, msaidizi wako wa Kiswahili. Niko hapa kukusaidia kujifunza na kuzungumza Kiswahili. **Ninaweza kukusaidiaje leo?** 😊"}],
+        height=600,
+        show_label=False,
+        avatar_images=(None, "user.png"),
+        bubble_full_width=False,
+        show_share_button=False,
+        type="messages"
     )
     
     msg = gr.Textbox(placeholder="Andika ujumbe wako hapa...", show_label=False)
     clear = gr.Button("Futa Mazungumzo")
 
-    def respond(message, chat_history):
-            # Add user message
-            chat_history.append({"role": "user", "content": message})
-            
-            # Generate response
-            bot_response = generate_response(message, chat_history)
-            
-            # Add assistant response
-            chat_history.append({"role": "assistant", "content": bot_response})
-            
-            return "", chat_history
+    def respond(message, history):
+        history.append({"role": "user", "content": message})
+        bot_response = generate_response(history)
+        history.append({"role": "assistant", "content": bot_response})
+        return "", history
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(lambda: [None, []], None, [msg, chatbot], queue=False)
 
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         auth=os.getenv("SPACE_AUTH"),
-        show_error=True
+        show_error=True,
+        share=False  # Explicitly disable sharing
     )