Update app.py

app.py

@@ -1,7 +1,7 @@
 import subprocess
 import sys
 
-# Self-install dependencies if missing
+# Self-install dependencies
 REQUIRED = ["gradio==4.44.1", "deep-translator==1.11.4", "llama-cpp-python==0.2.90"]
 for pkg in REQUIRED:
     try:
@@ -10,23 +10,21 @@ for pkg in REQUIRED:
         print(f" Installing {pkg}...")
         subprocess.check_call([sys.executable, "-m", "pip", "install", pkg])
 
-# NOW import normally
 import gradio as gr
 from llama_cpp import Llama
 from deep_translator import GoogleTranslator
 
-# 1. Load Potodoo (Optimized for HF Free Tier)
+# Load Potodoo
 print("🥔 Loading Potodoo...")
 llm = Llama(
-    model_path="potodoo_v1.gguf",  # ✅ FIXED: Matches your uploaded filename
-    n_ctx=512,                     # Reduced context to save RAM
-    n_gpu_layers=0,                # Force CPU (Free tier has no GPU)
-    n_threads=2,                   # Limit threads for stability
+    model_path="potodoo_v1.gguf",
+    n_ctx=512,
+    n_gpu_layers=0,
+    n_threads=2,
     verbose=False
 )
 print("✅ Potodoo ready!")
 
-# Language mapping for translation
 LANG_MAP = {
     "Português (BR)": "pt",
     "Español": "es",
@@ -36,28 +34,27 @@ LANG_MAP = {
     "한국어": "ko"
 }
 
-def chat(message, history, language):
-    # Convert Gradio history format to list of dicts
+def respond(message, history, language):
+    # Translate input to English
+    english_input = message
+    target_lang = LANG_MAP.get(language)
+    
+    if target_lang:
+        try:
+            t = GoogleTranslator(source=target_lang, target='en')
+            english_input = t.translate(message)
+        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]})
 
-    # Translate User Input -> English
-    english_input = message
-    target_lang_code = LANG_MAP.get(language)
-    
-    if target_lang_code:
-        try:
-            translator = GoogleTranslator(source=target_lang_code, target='en')
-            english_input = translator.translate(message)
-        except Exception as e:
-            print(f"Input translation error: {e}")
-
-    # Build Prompt with System Instruction
-    messages = [{"role": "system", "content": "You are Potodoo, a helpful and quirky AI assistant."}]
-    messages.extend(chat_history[-4:])  # Keep last 4 turns to save memory
+    messages = [{"role": "system", "content": "You are Potodoo, a helpful AI."}]
+    messages.extend(chat_history[-4:])
     messages.append({"role": "user", "content": english_input})
     
     prompt = ""
@@ -65,29 +62,24 @@ def chat(message, history, language):
         prompt += f"<|im_start|>{msg['role']}\n{msg['content']}<|im_end|>\n"
     prompt += "<|im_start|>assistant\n"
 
-    # Generate Response
+    # Generate
     output = llm(prompt, max_tokens=128, stop=["<|im_end|>"], temperature=0.7)
-    english_response = output['choices'][0]['text'].strip()
+    eng_resp = output['choices'][0]['text'].strip()
 
-    # Translate Response -> Target Language
-    final_response = english_response
-    if target_lang_code:
+    # Translate back
+    final_resp = eng_resp
+    if target_lang:
         try:
-            translator = GoogleTranslator(source='en', target=target_lang_code)
-            final_response = translator.translate(english_response)
-        except Exception as e:
-            print(f"Output translation error: {e}")
-
-    return final_response
+            t = GoogleTranslator(source='en', target=target_lang)
+            final_resp = t.translate(eng_resp)
+        except:
+            pass
 
-def respond(message, chat_history, language):
-    bot_message = chat(message, chat_history, language)
-    chat_history.append((message, bot_message))
-    return "", chat_history
+    return final_resp
 
-# 3. Build the Website UI
+# UI
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
-    gr.Markdown("#  Potodoo\n*Your tiny, multilingual AI assistant.*")
+    gr.Markdown("# 🥔 Potodoo\n*Multilingual AI Assistant*")
     
     chatbot = gr.Chatbot(height=400, type="tuples")
     
@@ -96,12 +88,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
         lang = gr.Dropdown(
             choices=["English", "Português (BR)", "Español", "Français", "Italiano", "日本語", "한국어"], 
             value="English", 
-            scale=1,
-            label="Language"
+            scale=1
         )
     
     msg.submit(respond, [msg, chatbot, lang], [msg, chatbot])
     gr.ClearButton([msg, chatbot], value="Clear Chat")
 
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name="0.0.0.0", server_port=7860)