Update app.py

app.py

@@ -1,5 +1,6 @@
 """
-Jajabor – SEBA Assamese Class 10 Tutor (Fixed for Hugging Face Spaces)
+Jajabor – SEBA Assamese Class 10 Tutor
+Simplified version for Hugging Face Spaces
 """
 
 import os
@@ -134,10 +135,11 @@ class SimpleTutor:
         self.embedding_model = None
         self.index = None
         self.corpus_chunks = []
-        self.loaded = False
         
+        print("🔄 Loading models...")
         self._load_models()
         self.load_pdfs()
+        print("✅ Tutor initialized")
     
     def _load_models(self):
         """Load models with error handling"""
@@ -153,35 +155,41 @@ class SimpleTutor:
                 self.llm = pipeline(
                     "text2text-generation",
                     model="google/flan-t5-small",
-                    device=-1,  # CPU
-                    torch_dtype="auto"
+                    device=-1
                 )
                 print("✅ LLM loaded")
             except Exception as e:
                 print(f"❌ Could not load LLM: {e}")
-        
-        self.loaded = True
     
     def load_pdfs(self):
         """Simple PDF loading"""
-        if not PDF_AVAILABLE or not os.path.exists(PDF_DIR):
-            print(f"PDF directory not found: {PDF_DIR}")
+        if not PDF_AVAILABLE:
+            print("📚 PDF reading not available")
+            return
+            
+        if not os.path.exists(PDF_DIR):
+            print(f"📁 PDF directory not found: {PDF_DIR}")
             return
         
         all_texts = []
-        for fname in os.listdir(PDF_DIR):
-            if fname.lower().endswith(".pdf"):
-                path = os.path.join(PDF_DIR, fname)
-                try:
-                    reader = PdfReader(path)
-                    text = ""
-                    for page in reader.pages:
-                        text += page.extract_text() or ""
-                    if text.strip():
-                        all_texts.append(text)
-                        print(f"📖 Loaded {fname}")
-                except Exception as e:
-                    print(f"Error reading {fname}: {e}")
+        pdf_files = [f for f in os.listdir(PDF_DIR) if f.lower().endswith('.pdf')]
+        
+        if not pdf_files:
+            print("📭 No PDF files found in directory")
+            return
+            
+        for fname in pdf_files:
+            path = os.path.join(PDF_DIR, fname)
+            try:
+                reader = PdfReader(path)
+                text = ""
+                for page in reader.pages:
+                    text += page.extract_text() or ""
+                if text.strip():
+                    all_texts.append(text)
+                    print(f"📖 Loaded {fname}")
+            except Exception as e:
+                print(f"❌ Error reading {fname}: {e}")
         
         # Simple text splitting
         self.corpus_chunks = []
@@ -189,11 +197,12 @@ class SimpleTutor:
             chunks = self._split_text(text)
             self.corpus_chunks.extend(chunks)
         
-        print(f"📚 Total chunks: {len(self.corpus_chunks)}")
+        print(f"📚 Total text chunks: {len(self.corpus_chunks)}")
         
         # Build FAISS index if we have chunks and embedding model
         if self.corpus_chunks and self.embedding_model and FAISS_AVAILABLE:
             try:
+                print("🔨 Building FAISS index...")
                 embs = self.embedding_model.encode(self.corpus_chunks, show_progress_bar=False).astype("float32")
                 dim = embs.shape[1]
                 self.index = faiss.IndexFlatL2(dim)
@@ -245,10 +254,7 @@ class SimpleTutor:
                 )
                 
                 if isinstance(response, list) and len(response) > 0:
-                    if hasattr(response[0], 'get'):
-                        answer = response[0].get('generated_text', 'উত্তৰ তৈয়াৰ কৰিব পৰা নগল।')
-                    else:
-                        answer = str(response[0])
+                    answer = response[0].get('generated_text', 'উত্তৰ তৈয়াৰ কৰিব পৰা নগল।')
                 else:
                     answer = str(response)
                     
@@ -353,21 +359,24 @@ def extract_text_from_image(image_path):
         print(f"OCR error: {e}")
         return ""
 
-# -------------------- GRADIO APP --------------------
-def main():
-    """Main function to run the app"""
+# -------------------- SIMPLE GRADIO APP --------------------
+def create_app():
+    """Create and return the Gradio app"""
     
     # Initialize components
+    print("🚀 Starting application...")
     init_db()
     tutor = SimpleTutor()
     
-    # Store user state in a simple way (avoiding gr.State issues)
+    # Store user state in memory (simple approach)
     user_states = {}
     
     def get_user_state(username):
         """Simple user state management"""
+        username = (username or "").strip()
         if not username:
             return None
+            
         if username not in user_states:
             user_id = get_or_create_user(username)
             if user_id:
@@ -383,14 +392,14 @@ def main():
             chat_history = []
         
         # Check if user is logged in
-        user_state = get_user_state(username.strip())
+        user_state = get_user_state(username)
         if not user_state:
-            new_history = chat_history + [[message, "⚠️ প্ৰথমে নাম লিখি লগিন কৰক।"]]
+            new_history = chat_history + [[message or "", "⚠️ প্ৰথমে নাম লিখি লগিন কৰক।"]]
             return new_history, ""
         
         # Combine text and image input
-        full_question = message.strip()
-        if image:
+        full_question = (message or "").strip()
+        if image is not None:
             ocr_text = extract_text_from_image(image)
             if ocr_text:
                 full_question += f"\n[ছবিৰ পাঠ: {ocr_text}]"
@@ -406,116 +415,101 @@ def main():
         log_interaction(user_state["user_id"], full_question, answer)
         
         # Update chat
-        display_question = message if message.strip() else "[ছবিৰ প্ৰশ্ন]"
+        display_question = message if message and message.strip() else "[ছবিৰ প্ৰশ্ন]"
         new_history = chat_history + [[display_question, answer]]
         return new_history, ""
     
     def clear_chat():
         """Clear chat history"""
-        return [], ""
+        return [], None
     
-    # Create Gradio interface
+    # Create Gradio interface - SIMPLIFIED without problematic components
     with gr.Blocks(
         title=APP_NAME,
-        theme=gr.themes.Soft(),
-        css="""
-        .container {
-            max-width: 1200px;
-            margin: auto;
-            padding: 20px;
-        }
-        .login-section {
-            background: #f8f9fa;
-            padding: 15px;
-            border-radius: 10px;
-            margin-bottom: 20px;
-        }
-        """
+        theme=gr.themes.Soft()
     ) as demo:
         
-        with gr.Column(elem_classes="container"):
-            gr.Markdown(f"# 🧭 {APP_NAME}")
-            gr.Markdown("SEBA Class 10 AI Tutor - Ask questions in Assamese or English")
+        gr.Markdown(f"# 🧭 {APP_NAME}")
+        gr.Markdown("SEBA Class 10 AI Tutor - Ask questions in Assamese or English")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### 👤 লগিন")
+                username = gr.Textbox(
+                    label="আপোনাৰ নাম",
+                    placeholder="আপোনাৰ নাম লিখক...",
+                    max_lines=1
+                )
+                gr.Markdown("""
+                ### 💡 টিপছ
+                - নাম লিখি প্ৰশ্ন সোধক
+                - পাঠ্যপুথিৰ PDF ফাইলসমূহ `pdfs/class10` ফ'ল্ডাৰত ৰাখক
+                - ছবি আপলোড কৰিলে OCR ৰ সহায়ত পাঠ পঢ়িব
+                """)
             
-            with gr.Row():
-                with gr.Column(scale=1):
-                    with gr.Group(elem_classes="login-section"):
-                        gr.Markdown("### 👤 লগিন")
-                        username = gr.Textbox(
-                            label="আপোনাৰ নাম",
-                            placeholder="আপোনাৰ নাম লিখক...",
-                            max_lines=1
-                        )
-                        gr.Markdown("""
-                        ### 💡 টিপছ
-                        - নাম লিখি প্ৰশ্ন সোধক
-                        - পাঠ্যপুথিৰ PDF ফাইলসমূহ `pdfs/class10` ফ'ল্ডাৰত ৰাখক
-                        - ছবি আপলোড কৰিলে OCR ৰ সহায়ত পাঠ পঢ়িব
-                        """)
+            with gr.Column(scale=2):
+                chatbot = gr.Chatbot(
+                    label="জাজাবৰ সৈতে কথোপকথন",
+                    height=500
+                )
                 
-                with gr.Column(scale=2):
-                    chatbot = gr.Chatbot(
-                        label="জাজাবৰ সৈতে কথোপকথন",
-                        height=500,
-                        show_copy_button=True
+                with gr.Row():
+                    message = gr.Textbox(
+                        label="প্ৰশ্ন",
+                        placeholder="আপোনাৰ প্ৰশ্ন ইয়াত লিখক...",
+                        lines=2
                     )
-                    
-                    with gr.Row():
-                        message = gr.Textbox(
-                            label="প্ৰশ্ন",
-                            placeholder="আপোনাৰ প্ৰশ্ন ইয়াত লিখক...",
-                            lines=2,
-                            scale=4
-                        )
-                    
-                    with gr.Row():
-                        image = gr.Image(
-                            label="ছবি আপলোড কৰক (ঐচ্ছিক)",
-                            type="filepath",
-                            height=150
-                        )
-                    
-                    with gr.Row():
-                        submit_btn = gr.Button("📤 প্ৰশ্ন পঠিয়াওক", variant="primary", scale=2)
-                        clear_btn = gr.Button("🧹 পৰিষ্কাৰ কৰক", variant="secondary", scale=1)
-            
-            # Event handlers
-            submit_btn.click(
-                fn=chat_function,
-                inputs=[message, image, chatbot, username],
-                outputs=[chatbot, message]
-            )
-            
-            message.submit(
-                fn=chat_function,
-                inputs=[message, image, chatbot, username],
-                outputs=[chatbot, message]
-            )
-            
-            clear_btn.click(
-                fn=clear_chat,
-                outputs=[chatbot, message]
-            )
+                
+                with gr.Row():
+                    image = gr.Image(
+                        label="ছবি আপলোড কৰক (ঐচ্ছিক)",
+                        type="filepath"
+                    )
+                
+                with gr.Row():
+                    submit_btn = gr.Button("📤 প্ৰশ্ন পঠিয়াওক", variant="primary")
+                    clear_btn = gr.Button("🧹 পৰিষ্কাৰ কৰক", variant="secondary")
+        
+        # Event handlers
+        submit_btn.click(
+            fn=chat_function,
+            inputs=[message, image, chatbot, username],
+            outputs=[chatbot, message]
+        )
+        
+        message.submit(
+            fn=chat_function,
+            inputs=[message, image, chatbot, username],
+            outputs=[chatbot, message]
+        )
+        
+        clear_btn.click(
+            fn=clear_chat,
+            outputs=[chatbot, image]
+        )
     
     return demo
 
-# -------------------- LAUNCH --------------------
+# -------------------- MAIN --------------------
 if __name__ == "__main__":
     if not GRADIO_AVAILABLE:
-        print("Gradio not available. Please install gradio.")
+        print("❌ Gradio not available. Please install gradio.")
         exit(1)
     
-    demo = main()
-    
-    # For Hugging Face Spaces, use share=False and don't specify server_name
     try:
+        demo = create_app()
+        print("✅ App created successfully")
+        
+        # Simple launch for Hugging Face Spaces
         demo.launch(
-            server_name="0.0.0.0" if os.getenv('SPACE_ID') else None,
+            server_name="0.0.0.0",
             server_port=7860,
-            share=False,  # Important: set to False for Spaces
-            show_error=True
+            share=False  # Critical: must be False for Spaces
         )
     except Exception as e:
-        print(f"Launch error: {e}")
-        # Fallback launch without server_name
-        demo.launch(share=False)
+        print(f"❌ Error launching app: {e}")
+        # Final fallback
+        try:
+            demo.launch(share=False)
+        except:
+            print("💥 Failed to launch application")