Update app.py

app.py

@@ -9,16 +9,13 @@ from PIL import Image
 
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# Safe client initialization
 try:
     if not GROQ_API_KEY:
-        print("⚠️ GROQ_API_KEY is missing! Check Secrets.")
         client = None
     else:
         client = Groq(api_key=GROQ_API_KEY)
-except Exception as e:
+except Exception:
     client = None
-    print(f"Error initializing Groq client: {e}")
 
 DEFAULT_MODEL = "llama-3.1-8b-instant"
 
@@ -50,96 +47,73 @@ def image_to_ascii(image_path, output_width=100):
         return f"[Error: {e}]"
 
 # =========================
-#  CHAT FUNCTION (UNIVERSAL COMPATIBILITY)
+#  CHAT LOGIC
 # =========================
 
 def chat_fn(message, history, mode, model, image, include_ascii):
-    try:
-        if not client:
-            return "⚠️ Error: Groq client not initialized."
-
-        # 1. Prepare System Prompt
-        system_prompt = MODE_PROMPTS.get(mode, MODE_PROMPTS["Normal Chat"])
-        messages = [{"role": "system", "content": str(system_prompt).strip()}]
-
-        # 2. MANUAL CONVERSION: Old Gradio (Tuples) -> Groq (Dicts)
-        # History format received: [['User Hi', 'Bot Hello'], ['User How are you', 'Bot Good']]
-        for pair in history:
-            if isinstance(pair, (list, tuple)) and len(pair) == 2:
-                user_h, bot_h = pair
-                if user_h:
-                    messages.append({"role": "user", "content": str(user_h)})
-                if bot_h:
-                    messages.append({"role": "assistant", "content": str(bot_h)})
-
-        # 3. Handle Current Message / Image
-        safe_message = str(message).strip() if message else ""
-        
-        if image and include_ascii:
-            try:
-                # Gradio passes image path string
-                ascii_art = image_to_ascii(image)
-                final_content = f"[ASCII Art from Uploaded Image]\n{ascii_art}\n\nUser message: {safe_message}"
-            except Exception:
-                final_content = safe_message
-        else:
-            final_content = safe_message
-
-        if not final_content:
-            return "⚠️ Error: Message is empty."
-
-        messages.append({"role": "user", "content": final_content})
-
-        # 4. Call Groq
-        response = client.chat.completions.create(
-            model=model,
-            messages=messages
-        )
-        return response.choices[0].message.content
+    if not client:
+        return "⚠️ Error: Groq API Key is missing in Settings > Secrets."
+
+    # 1. Build the messages list for Groq API
+    system_prompt = MODE_PROMPTS.get(mode, MODE_PROMPTS["Normal Chat"])
+    messages = [{"role": "system", "content": system_prompt}]
+
+    # 2. Convert Gradio's history (list of lists) to Groq's format (list of dicts)
+    if history:
+        for user_msg, bot_msg in history:
+            if user_msg: messages.append({"role": "user", "content": str(user_msg)})
+            if bot_msg: messages.append({"role": "assistant", "content": str(bot_msg)})
+
+    # 3. Process current input
+    user_input_text = str(message).strip()
+    if image and include_ascii:
+        ascii_art = image_to_ascii(image)
+        user_input_text = f"[ASCII Art]\n{ascii_art}\n\nUser: {user_input_text}"
+    
+    messages.append({"role": "user", "content": user_input_text})
 
+    # 4. Get response
+    try:
+        completion = client.chat.completions.create(model=model, messages=messages)
+        return completion.choices[0].message.content
     except Exception as e:
-        return f"⚠️ Error: {str(e)}"
+        return f"⚠️ Groq Error: {str(e)}"
 
 # =========================
-#  UI (COMPATIBLE MODE)
+#  STABLE UI
 # =========================
 
-with gr.Blocks(title="NeoHelper") as demo:
-    gr.Markdown("## 🦙 NeoHelper — Eyad’s Assistant")
+with gr.Blocks() as demo:
+    gr.Markdown("# 🦙 NeoHelper")
     
     with gr.Row():
-        mode_dd = gr.Dropdown(list(MODE_PROMPTS.keys()), value="Normal Chat", label="Mode")
-        model_dd = gr.Dropdown([
-            "llama-3.1-8b-instant", 
-            "llama-3.1-70b-versatile", 
-            "mixtral-8x7b-instruct",
-            "gemma2-9b-it"
-        ], value=DEFAULT_MODEL, label="Model")
+        mode_dd = gr.Dropdown(choices=list(MODE_PROMPTS.keys()), value="Normal Chat", label="Mode")
+        model_dd = gr.Dropdown(choices=["llama-3.1-8b-instant", "llama-3.1-70b-versatile"], value=DEFAULT_MODEL, label="Model")
 
-    # FIX: Removed 'type="messages"' to support older Gradio versions
-    chatbot = gr.Chatbot(height=500) 
+    # We leave the chatbot as default. No type="messages" and no complex settings.
+    chatbot = gr.Chatbot(height=450)
 
     with gr.Row():
-        user_input = gr.Textbox(placeholder="Message...", show_label=False, lines=2)
-        send_btn = gr.Button("Send")
+        user_input = gr.Textbox(placeholder="Ask anything...", show_label=False, scale=4)
+        send_btn = gr.Button("Send", variant="primary", scale=1)
 
-    image_input = gr.Image(type="filepath", label="Upload Image (Optional)")
-    ascii_toggle = gr.Checkbox(label="Convert image to ASCII", value=False)
+    image_input = gr.Image(type="filepath", label="Image (Optional)")
+    ascii_toggle = gr.Checkbox(label="Convert image to ASCII art")
 
-    def on_send(msg, history, mode, model, img, ascii_on):
-        if history is None: history = []
-        
-        # Get response string from Groq
-        reply = chat_fn(msg, history, mode, model, img, ascii_on)
+    def respond(msg, chat_history, mode, model, img, ascii_on):
+        # 1. Get the bot's response
+        bot_message = chat_fn(msg, chat_history, mode, model, img, ascii_on)
         
-        # FIX: Append as a TUPLE [User, Bot] for the UI
-        history.append([str(msg), str(reply)])
+        # 2. Append the new interaction as a LIST [User, Bot]
+        # This is the "Classic" format that works on all Gradio versions.
+        chat_history.append([msg, bot_message])
         
-        return history, ""
+        # 3. Return history and clear the input box
+        return chat_history, ""
 
     send_btn.click(
-        on_send,
-        inputs=[user_input, chatbot, mode_dd, model_dd, image_input, ascii_toggle],
+        respond, 
+        inputs=[user_input, chatbot, mode_dd, model_dd, image_input, ascii_toggle], 
         outputs=[chatbot, user_input]
     )