Update src/streamlit_app.py

src/streamlit_app.py

@@ -150,6 +150,37 @@ def generate_response(pipeline_obj, prompt, system_prompt, max_length=256, tempe
     except Exception as e:
         return f"❌ Error generating response: {str(e)}"
 
+def create_fallback_demo():
+    """Create a simple demo mode when model loading fails"""
+    st.warning("🔧 Model loading failed. Running in demo mode with simulated responses.")
+    
+    class DemoTokenizer:
+        def __init__(self):
+            self.eos_token_id = 2
+    
+    class DemoPipeline:
+        def __init__(self):
+            self.tokenizer = DemoTokenizer()
+        
+        def __call__(self, prompt, **kwargs):
+            # Simulate response generation
+            time.sleep(1)  # Simulate processing time
+            
+            # Simple demo responses based on input
+            if any(arabic_word in prompt for arabic_word in ['مرحبا', 'السلام', 'أهلا']):
+                response = "مرحبا بك! أنا حكيم، مساعدك الذكي. كيف يمكنني مساعدتك اليوم؟"
+            elif 'hello' in prompt.lower() or 'hi' in prompt.lower():
+                response = "Hello! I'm Hakim, your AI assistant. How can I help you today?"
+            elif 'what' in prompt.lower() and 'ai' in prompt.lower():
+                response = "Artificial Intelligence (AI) refers to computer systems that can perform tasks that typically require human intelligence, such as learning, reasoning, and problem-solving."
+            else:
+                response = "I understand your question. This is a demo response since the actual model couldn't be loaded. In a real deployment, I would provide a more detailed and contextual answer based on the Rabe3/Hakim model."
+            
+            return [{'generated_text': response}]
+    
+    st.info("✅ Demo mode initialized!")
+    return DemoTokenizer(), None, DemoPipeline()
+
 def show_requirements_info():
     """Show information about missing requirements"""
     st.error("🚫 Required libraries are missing!")