Update app.py

app.py

@@ -10,50 +10,20 @@ from sentence_transformers import SentenceTransformer
 import faiss
 import json
 import numpy as np
-import os
-import time
 
 # Load model and index at startup
 print("Loading model and index...")
 MODEL_NAME = 'intfloat/multilingual-e5-large'
+model = SentenceTransformer(MODEL_NAME)
 
-try:
-    model = SentenceTransformer(MODEL_NAME)
-    print(f"Model loaded: {MODEL_NAME}")
-    
-    # Load FAISS index
-    index_path = 'faiss_index.bin'
-    if not os.path.exists(index_path):
-        # Try to find the file with different naming
-        for file in os.listdir('.'):
-            if file.endswith('.bin') and 'faiss' in file.lower():
-                index_path = file
-                break
-    
-    index = faiss.read_index(index_path)
-    print(f"FAISS index loaded with {index.ntotal} vectors")
-    
-    # Load metadata
-    metadata_path = 'metadata.json'
-    if not os.path.exists(metadata_path):
-        # Try to find the file with different naming
-        for file in os.listdir('.'):
-            if file.endswith('.json') and 'metadata' in file.lower():
-                metadata_path = file
-                break
-    
-    with open(metadata_path, 'r', encoding='utf-8') as f:
-        metadata = json.load(f)
-    
-    print(f"Loaded {len(metadata):,} question-answer pairs")
-    
-except Exception as e:
-    print(f"Error loading resources: {e}")
-    # Create dummy data for demo purposes
-    model = None
-    index = None
-    metadata = []
-    print("Running in demo mode with limited functionality")
+# Load FAISS index
+index = faiss.read_index('faiss_index.bin')
+
+# Load metadata
+with open('metadata.json', 'r', encoding='utf-8') as f:
+    metadata = json.load(f)
+
+print(f"Loaded {len(metadata):,} question-answer pairs")
 
 def search_answers(query, top_k=1):
     """
@@ -66,350 +36,213 @@ def search_answers(query, top_k=1):
     Returns:
         List of top matching answers
     """
-    if model is None or index is None or not metadata:
-        # Return demo response
-        return [{
-            'rank': 1,
-            'question': "Demo mode: System is initializing",
-            'answer': "Sɛ moadwendwen sɛ, twerɛ wo nsɛm bio anaasɛ san bra bio.\n\nPlease try again or rephrase your question.",
-            'similarity': 0.8
-        }]
+    # Encode query with E5 prefix for query encoding
+    query_text = "query: " + query.strip()
+    query_embedding = model.encode(
+        [query_text],
+        normalize_embeddings=True
+    )
     
-    try:
-        # Encode query with E5 prefix for query encoding
-        query_text = "query: " + query.strip()
-        query_embedding = model.encode(
-            [query_text],
-            normalize_embeddings=True,
-            show_progress_bar=False
-        )
-        
-        # Search in FAISS
-        distances, indices = index.search(query_embedding.astype('float32'), top_k)
-        
-        # Prepare results
-        results = []
-        for idx, (similarity, index_pos) in enumerate(zip(distances[0], indices[0]), 1):
-            if 0 <= index_pos < len(metadata):
-                item = metadata[index_pos]
-                results.append({
-                    'rank': idx,
-                    'question': item.get('question', 'No question found'),
-                    'answer': item.get('answer', 'No answer found'),
-                    'similarity': float(similarity)
-                })
-        
-        # If no results found, return a default response
-        if not results:
+    # Search in FAISS
+    distances, indices = index.search(query_embedding.astype('float32'), top_k)
+    
+    # Prepare results
+    results = []
+    for idx, (similarity, index_pos) in enumerate(zip(distances[0], indices[0]), 1):
+        if index_pos < len(metadata):
+            item = metadata[index_pos]
             results.append({
-                'rank': 1,
-                'question': "No exact match found",
-                'answer': "Menya mmuae a ɛne wo nsɛmmisa no hyia. Yɛsrɛ wo sɛ bɔ mmɔden bio wɔ nsɛmmisa foforɔ so anaasɛ kyerɛw wo nsɛmmisa no bio wɔ Twi anaa English mu.\n\nI couldn't find an exact match. Please try a different question or rephrase your question in Twi or English.",
-                'similarity': 0.0
+                'rank': idx,
+                'question': item['question'],
+                'answer': item['answer'],
+                'similarity': float(similarity)
             })
-        
-        return results
     
-    except Exception as e:
-        print(f"Search error: {e}")
-        return [{
-            'rank': 1,
-            'question': "Error in search",
-            'answer': "Sɛ moadwendwen sɛ, adwenemhye a ɛwɔ sistɛm mu no ama adwuma no atɔe. Yɛsrɛ wo sɛ san bɛsɛ bio.\n\nThere was an error processing your request. Please try again.",
-            'similarity': 0.0
-        }]
+    return results
 
-# Welcome message in Twi - FIXED: Removed extra quotes
+# Welcome message in Twi
 WELCOME_MESSAGE = """Akwaaba! 👋
 
 Meyɛ Twi Akwahosan Boafoɔ. Metumi aboa wo wɔ nsɛmmisa ahorow a ɛfa akwahosan ho.
 
 Bisa me nsɛmmisa biara wɔ Twi anaa English mu, na mɛma wo mmuae wɔ Twi mu.
 
-**Nhwɛsoɔ Nsɛmmisa:**
-- "Dɛn na menyɛ fa alcohol ho wɔ nufunom bere mu?"
-- "Ɛyɛ dɛn sɛ meyi nyinsɛn adi ntɛm?"
-- "What should I do if I have malaria?"
-
 ---
 
-_Welcome! I'm your Twi Health Assistant. I can help you with various health-related questions. Ask me anything in Twi or English, and I'll provide answers in Twi._
-
-**Example Questions:**
-- "What should I do about alcohol while breastfeeding?"
-- "How do I know if I'm pregnant early?"
-- "What are the symptoms of malaria?" """
-
-def user_message(user_msg, history):
-    """Handle user message submission"""
-    if not user_msg or user_msg.strip() == "":
-        return "", history
-    
-    # Add user message to history
-    if history is None:
-        history = []
-    
-    history.append([user_msg, None])
-    return "", history
-
-def bot_message(history):
-    """Generate bot response"""
-    if not history or history[-1][0] is None:
-        return history
-    
-    user_msg = history[-1][0]
-    
-    # Get bot response
-    results = search_answers(user_msg, top_k=1)
-    
-    if results:
-        bot_response = results[0]['answer']
-        
-        # Add confidence indicator if similarity is low
-        if results[0]['similarity'] < 0.3:
-            bot_response = "**Akyiakyia:** Menya mmuae a ɛne wo nsɛmmisa no hyia koraa no. Saa nti, me de mmuae a ɛte saa na ɛbɛtumi aboa wo.\n\n" + bot_response
-    else:
-        bot_response = "Menya mmuae biara a ɛne wo nsɛmmisa no hyia. Yɛsrɛ wo sɛ bɔ mmɔden bio wɔ nsɛmmisa foforɔ so anaa kyerɛw wo nsɛmmisa no bio.\n\n_I couldn't find an answer to your question. Please try rephrasing or asking a different question._"
-    
-    history[-1][1] = bot_response
-    return history
-
-def clear_chat():
-    """Reset chat to welcome message"""
-    return [[None, WELCOME_MESSAGE]]
+_Welcome! I'm your Twi Health Assistant. I can help you with various health-related questions. Ask me anything in Twi or English, and I'll provide answers in Twi._"""
+
+# Create custom CSS for ChatGPT-like appearance
+custom_css = """
+#chatbot {
+    height: 600px;
+    overflow-y: auto;
+}
+
+.message {
+    padding: 12px 16px !important;
+    border-radius: 18px !important;
+    margin: 8px 0 !important;
+}
+
+#input-box textarea {
+    border-radius: 24px !important;
+}
+
+.contain {
+    max-width: 900px !important;
+    margin: auto !important;
+}
+
+footer {
+    display: none !important;
+}
+
+.chat-header {
+    text-align: center;
+    padding: 20px;
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    border-radius: 12px;
+    margin-bottom: 20px;
+}
+"""
 
 # Create Gradio interface with ChatGPT style
 with gr.Blocks(
+    css=custom_css,
     theme=gr.themes.Soft(
         primary_hue="purple",
         secondary_hue="blue",
-        font=["Inter", "system-ui", "sans-serif"]
+        font=gr.themes.GoogleFont("Inter")
     ),
-    title="Twi Akwahosan Boafoɔ | Twi Health Assistant",
-    css="""
-    #chatbot {
-        min-height: 400px;
-        max-height: 600px;
-        overflow-y: auto;
-        border: 1px solid #e0e0e0;
-        border-radius: 10px;
-        padding: 15px;
-        background: #f9f9f9;
-    }
-    
-    .message {
-        padding: 12px 16px;
-        border-radius: 18px;
-        margin: 8px 0;
-        max-width: 80%;
-    }
-    
-    .user-message {
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        margin-left: auto;
-    }
-    
-    .bot-message {
-        background: #ffffff;
-        border: 1px solid #e0e0e0;
-        color: #333;
-    }
-    
-    .input-container {
-        border-top: 1px solid #e0e0e0;
-        padding-top: 20px;
-        margin-top: 20px;
-    }
-    
-    .header {
-        text-align: center;
-        padding: 20px;
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        border-radius: 12px;
-        margin-bottom: 20px;
-    }
-    
-    .footer {
-        text-align: center;
-        margin-top: 20px;
-        padding: 10px;
-        color: #666;
-        font-size: 0.9em;
-        border-top: 1px solid #e0e0e0;
-    }
-    
-    .examples {
-        background: #f0f0f0;
-        padding: 15px;
-        border-radius: 10px;
-        margin: 10px 0;
-    }
-    
-    .example-btn {
-        margin: 5px;
-        background: #e0e0e0;
-        border: none;
-        border-radius: 20px;
-        padding: 5px 15px;
-        cursor: pointer;
-        transition: background 0.3s;
-    }
-    
-    .example-btn:hover {
-        background: #d0d0d0;
-    }
-    """
+    title="Twi Akwahosan Boafoɔ | Twi Health Assistant"
 ) as demo:
     
     # Header
     gr.HTML("""
-        <div class="header">
-            <h1 style="margin:0; font-size: 2em; color: white;">🏥 Twi Akwahosan Boafoɔ</h1>
-            <p style="margin:10px 0 0 0; opacity: 0.9; color: white;">Twi Health Information Assistant</p>
-            <p style="margin:5px 0 0 0; font-size: 0.9em; opacity: 0.8; color: white;">
-                Ask health questions in any language, get answers in Twi
-            </p>
+        <div class="chat-header">
+            <h1 style="margin:0; font-size: 2em;">🏥 Twi Akwahosan Boafoɔ</h1>
+            <p style="margin:10px 0 0 0; opacity: 0.9;">Twi Health Information Assistant</p>
         </div>
     """)
     
-    # Chatbot interface
+    # Chatbot interface - simplified without avatar_images to avoid bug
     chatbot = gr.Chatbot(
         value=[[None, WELCOME_MESSAGE]],
         elem_id="chatbot",
         label="",
         show_label=False,
-        height=500
+        bubble_full_width=False,
+        height=600
     )
     
-    # Chat input section
+    # Input section
     with gr.Row():
-        with gr.Column(scale=9):
-            msg = gr.Textbox(
-                placeholder="Kyerɛw wo nsɛmmisa wɔ ha... | Type your question here...",
-                show_label=False,
-                container=False,
-                lines=2,
-                elem_id="chat-input"
-            )
-        with gr.Column(scale=1):
-            submit_btn = gr.Button("📤", variant="primary", size="lg")
+        msg = gr.Textbox(
+            placeholder="Kyerɛw wo nsɛmmisa wɔ ha... | Type your question here...",
+            show_label=False,
+            scale=9,
+            container=False,
+            elem_id="input-box"
+        )
+        submit_btn = gr.Button("📤", scale=1, variant="primary", size="lg")
     
-    # Control buttons
+    # Clear button
     with gr.Row():
-        clear_btn = gr.Button("🗑️ Firi Aseɛ Bio", variant="secondary")
-        gr.HTML('<div style="flex-grow: 1;"></div>')
-        gr.Markdown("**System Status:** 🟢 Ready" if model else "**System Status:** 🟡 Limited Mode")
+        clear_btn = gr.Button("🗑️ Firi Aseɛ Bio | Start New Chat", size="sm")
     
-    # Example questions
+    # Example questions section
     with gr.Accordion("📝 Nhwɛsoɔ Nsɛmmisa | Example Questions", open=False):
-        examples_html = """
-        <div class="examples">
-            <p><strong>Click any example to try:</strong></p>
-            <div style="display: flex; flex-wrap: wrap; gap: 8px;">
-        """
+        gr.Examples(
+            examples=[
+                ["Dɛn na menyɛ fa alcohol ho wɔ nufunom bere mu?"],
+                ["Ɛyɛ dɛn sɛ meyi nyinsɛn adi ntɛm?"],
+                ["Mɛnya ambulance service frɛ nɔma no?"],
+                ["What should I do if I have malaria?"],
+                ["Dɛn na memfa ahwɛ me tirim yadeɛ?"],
+                ["How do I take care of a newborn baby?"],
+                ["Nsuo dodow bɛn na ɛsɛ sɛ menom da biara?"],
+            ],
+            inputs=msg,
+            label=""
+        )
+    
+    # Information section
+    with gr.Accordion("ℹ️ Nkyerɛkyerɛmu | About This Assistant", open=False):
+        gr.Markdown("""
+        ### Sɛdeɛ Ɛyɛ Adwuma | How It Works
         
-        example_questions = [
-            "Dɛn na menyɛ fa alcohol ho wɔ nufunom bere mu?",
-            "Ɛyɛ dɛn sɛ meyi nyinsɛn adi ntɛm?",
-            "Mɛnya ambulance service frɛ nɔma no?",
-            "What should I do if I have malaria?",
-            "Dɛn na memfa ahwɛ me tirim yadeɛ?",
-            "How do I take care of a newborn baby?",
-            "Nsuo dodow bɛn na ɛsɛ sɛ menom da biara?",
-            "Pregnancy symptoms in first month",
-            "Breastfeeding tips for new mothers",
-            "Postpartum depression symptoms"
-        ]
+        1. **Kyerɛw wo nsɛmmisa** wɔ Twi anaa English mu
+        2. **Sistɛm no bɛhwehwɛ** nsɛmmisa a ɛne wo de no hyia wɔ database no mu
+        3. **Wonya mmuae** wɔ Twi mu ntɛm ara
         
-        for i, question in enumerate(example_questions):
-            # Escape single quotes and backticks for JavaScript
-            safe_question = question.replace("'", "\\'").replace("`", "\\`")
-            examples_html += f'<button class="example-btn" onclick="document.querySelector(\'#chat-input textarea\').value = \'{safe_question}\'; document.querySelector(\'#chat-input textarea\').dispatchEvent(new Event(\'input\', {{bubbles: true}}));">{question[:40]}{"..." if len(question) > 40 else ""}</button>'
+        ### 📊 Technical Details
         
-        examples_html += "</div></div>"
-        gr.HTML(examples_html)
-    
-    # Information section
-    with gr.Accordion("ℹ️ Nkyerɛkyerɛmu | Information", open=False):
-        info_text = f"""
-        ### 📊 Technical Information
+        - **Model**: Multilingual E5-Large (Semantic Search)
+        - **Database**: Health Q&A pairs in Twi
+        - **Retrieval**: FAISS vector similarity search
+        - **Languages**: Primarily Twi, with English support
         
-        - **Embedding Model**: Multilingual E5-Large
-        - **Search Engine**: FAISS vector database
-        - **Index Size**: {len(metadata):,} Q&A pairs
-        - **Supported Languages**: 100+ languages including Twi, English, Ga, Ewe, Hausa
+        ### 🔒 Ahobammɔ | Privacy
         
-        ### 🔒 Privacy & Security
+        Wo nsɛmmisa ahorow no yɛ ahobammɔ na wɔmfa nkora so.
         
-        - Your questions are processed in real-time
-        - No data is stored permanently
-        - All processing happens on the server
+        _Your questions are private and not stored permanently._
         
-        ### ⚠️ Important Notice
+        ### ⚠️ Kɔkɔbɔ | Important Notice
         
-        **Disclaimer**: This system provides health information for educational purposes only. Always consult with qualified healthcare professionals for medical advice.
+        Sɛ ɛyɛ emergency anaa yadeɛ a emu yɛ den a, kɔ ayaresabea anaa frɛ dokta.
         
-        **For emergencies**: Please call 112 (Ghana emergency number) or visit the nearest hospital.
-        """
-        gr.Markdown(info_text)
+        _For emergencies or serious health conditions, please visit a hospital or call a doctor._
+        """)
+    
+    # Event handlers
+    def user_message(user_msg, history):
+        """Handle user message submission"""
+        return "", history + [[user_msg, None]]
+    
+    def bot_message(history):
+        """Generate bot response"""
+        if history[-1][1] is not None:
+            return history
+        
+        user_msg = history[-1][0]
+        
+        # Get bot response
+        results = search_answers(user_msg, top_k=1)
+        
+        if results and results[0]['similarity'] > 0.3:
+            bot_response = results[0]['answer']
+        else:
+            bot_response = "Menya mmuae biara a ɛne wo nsɛmmisa no hyia. Yɛsrɛ wo sɛ bɔ mmɔden bio wɔ nsɛmmisa foforɔ so anaa kyerɛw wo nsɛmmisa no bio.\n\n_Menya mmuae a ɛne wo nsɛmmisa no hyia. Sɛ wobɛtumi akyerɛw wo nsɛmmisa no bio anaa bisa nsɛmmisa foforɔ a, ɛbɛboa._"
+        
+        history[-1][1] = bot_response
+        return history
     
-    # Footer
-    gr.HTML("""
-    <div class="footer">
-        <p>🇬🇭 Built for the Ghanaian Community | Powered by Multilingual E5 & FAISS</p>
-        <p style="font-size: 0.8em; margin-top: 5px;">
-            <em>"Akwaaba" means "Welcome" in Twi | Medaase pa ara! 🙏</em>
-        </p>
-    </div>
-    """)
+    def clear_chat():
+        """Reset chat to welcome message"""
+        return [[None, WELCOME_MESSAGE]]
     
-    # Event handlers with proper queue management
-    msg.submit(
-        fn=user_message,
-        inputs=[msg, chatbot],
-        outputs=[msg, chatbot],
-        queue=True
-    ).then(
-        fn=bot_message,
-        inputs=[chatbot],
-        outputs=[chatbot],
-        queue=True
+    # Wire up events
+    msg.submit(user_message, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot_message, chatbot, chatbot
     )
     
-    submit_btn.click(
-        fn=user_message,
-        inputs=[msg, chatbot],
-        outputs=[msg, chatbot],
-        queue=True
-    ).then(
-        fn=bot_message,
-        inputs=[chatbot],
-        outputs=[chatbot],
-        queue=True
+    submit_btn.click(user_message, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot_message, chatbot, chatbot
     )
     
-    clear_btn.click(
-        fn=clear_chat,
-        inputs=None,
-        outputs=[chatbot],
-        queue=True
-    )
+    clear_btn.click(clear_chat, None, chatbot, queue=False)
+    
+    # Footer
+    gr.Markdown("""
+    ---
+    <div style="text-align: center; color: #666; font-size: 0.9em;">
+        <p>Powered by Multilingual E5 & FAISS | Yɛde Twi kasa reba anim 🇬🇭</p>
+    </div>
+    """)
 
-# Launch the app with proper settings for Hugging Face Spaces
+# Launch the app
 if __name__ == "__main__":
-    # Check if running in Hugging Face Spaces
-    is_spaces = os.getenv('SPACE_ID') is not None
-    
-    # Launch with appropriate settings - FIXED: Set share=True for HF Spaces
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True,  # Enable share to avoid localhost issues
-        show_error=True,
-        debug=False,
-        show_api=False,  # Disable API to avoid JSON schema issues
-        quiet=True,  # Reduce console output
-        enable_queue=True  # Enable queue for better stability
-    )
+    demo.queue()
+    demo.launch()