Update app.py

app.py

@@ -1,29 +1,50 @@
 import subprocess
 import sys
+import os
 
-# Self-install dependencies
+# Self-install dependencies with error handling
 REQUIRED = ["gradio==4.44.1", "deep-translator==1.11.4", "llama-cpp-python==0.2.90"]
 for pkg in REQUIRED:
     try:
         __import__(pkg.split("==")[0])
     except ImportError:
         print(f" Installing {pkg}...")
-        subprocess.check_call([sys.executable, "-m", "pip", "install", pkg])
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir", pkg])
 
 import gradio as gr
-from llama_cpp import Llama
 from deep_translator import GoogleTranslator
 
-# Load Potodoo
+# Check if llama_cpp imported successfully
+try:
+    from llama_cpp import Llama
+except ImportError as e:
+    print(f" FATAL: llama_cpp failed to import: {e}")
+    print(" This usually means it was compiled without AVX2 support.")
+    print(" Trying to reinstall with CMAKE_ARGS...")
+    os.environ["CMAKE_ARGS"] = "-DLLAMA_BLAS=OFF -DLLAMA_AVX2=ON"
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir", "--force-reinstall", "llama-cpp-python==0.2.90"])
+    from llama_cpp import Llama
+
+# Verify model file exists
+MODEL_PATH = "potodoo_v1.gguf"
+if not os.path.exists(MODEL_PATH):
+    print(f" FATAL: Model file '{MODEL_PATH}' not found!")
+    print(f" Files in /app: {os.listdir('/app')}")
+    raise FileNotFoundError(f"Model file missing: {MODEL_PATH}")
+
 print("🥔 Loading Potodoo...")
-llm = Llama(
-    model_path="potodoo_v1.gguf",
-    n_ctx=512,
-    n_gpu_layers=0,
-    n_threads=2,
-    verbose=False
-)
-print("✅ Potodoo ready!")
+try:
+    llm = Llama(
+        model_path=MODEL_PATH,
+        n_ctx=512,
+        n_gpu_layers=0,
+        n_threads=2,
+        verbose=True  # Enable verbose to see actual errors
+    )
+    print("✅ Potodoo ready!")
+except Exception as e:
+    print(f" FATAL: Failed to load model: {e}")
+    raise
 
 LANG_MAP = {
     "Português (BR)": "pt",
@@ -35,7 +56,6 @@ LANG_MAP = {
 }
 
 def respond(message, history, language):
-    # Translate input to English
     english_input = message
     target_lang = LANG_MAP.get(language)
     
@@ -46,14 +66,13 @@ def respond(message, history, language):
         except:
             pass
 
-    # Build prompt
     chat_history = []
     if history:
         for turn in history:
             chat_history.append({"role": "user", "content": turn[0]})
             chat_history.append({"role": "assistant", "content": turn[1]})
 
-    messages = [{"role": "system", "content": "You are Potodoo, a helpful AI."}]
+    messages = [{"role": "system", "content": "You are Potodoo."}]
     messages.extend(chat_history[-4:])
     messages.append({"role": "user", "content": english_input})
     
@@ -62,11 +81,9 @@ def respond(message, history, language):
         prompt += f"<|im_start|>{msg['role']}\n{msg['content']}<|im_end|>\n"
     prompt += "<|im_start|>assistant\n"
 
-    # Generate
     output = llm(prompt, max_tokens=128, stop=["<|im_end|>"], temperature=0.7)
     eng_resp = output['choices'][0]['text'].strip()
 
-    # Translate back
     final_resp = eng_resp
     if target_lang:
         try:
@@ -77,21 +94,18 @@ def respond(message, history, language):
 
     return final_resp
 
-# UI
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
-    gr.Markdown("# 🥔 Potodoo\n*Multilingual AI Assistant*")
-    
+    gr.Markdown("# 🥔 Potodoo")
     chatbot = gr.Chatbot(height=400, type="tuples")
     
     with gr.Row():
-        msg = gr.Textbox(scale=4, show_label=False, placeholder="Type a message...")
+        msg = gr.Textbox(scale=4, show_label=False, placeholder="Type...")
         lang = gr.Dropdown(
             choices=["English", "Português (BR)", "Español", "Français", "Italiano", "日本語", "한국어"], 
-            value="English", 
-            scale=1
+            value="English", scale=1
         )
     
     msg.submit(respond, [msg, chatbot, lang], [msg, chatbot])
-    gr.ClearButton([msg, chatbot], value="Clear Chat")
+    gr.ClearButton([msg, chatbot])
 
 demo.launch(server_name="0.0.0.0", server_port=7860)