Update app.py

app.py

@@ -12,14 +12,22 @@ def chat_engine(message, history, model_choice):
     pipe = pipeline("text-generation", model=MODELS[model_choice])
     
     prompt = f"System: You are {model_choice}, the flagship AI of Zakomako4567. Use clean code.\n"
-    for user_msg, ai_msg in history:
-        prompt += f"User: {user_msg}\nAI: {ai_msg}\n"
+    
+    # Updated history handling for Gradio 6.0 compatibility
+    for turn in history:
+        user_msg = turn['role'] == 'user'
+        content = turn['content']
+        if user_msg:
+            prompt += f"User: {content}\n"
+        else:
+            prompt += f"AI: {content}\n"
+            
     prompt += f"User: {message}\nAI:"
     
     res = pipe(prompt, max_new_tokens=512, do_sample=True, temperature=0.7)[0]['generated_text']
     return res.split("AI:")[-1].strip()
 
-# CSS for a "Premium" Interface
+# Premium CSS Styling
 custom_css = """
 footer {visibility: hidden}
 .gradio-container { background: #050505 !important; color: white !important; }
@@ -36,13 +44,11 @@ footer {visibility: hidden}
 """
 
 # 2. Building the Interface
-# We removed theme/css from here to satisfy the Gradio 6.0 warning
 with gr.Blocks() as demo:
     
-    # Replaced gr.Div with gr.Column to avoid the AttributeError
     with gr.Column(elem_id="title-header"):
         gr.Markdown(f"# 🚀 ZewAI 3 by @Zakomako4567")
-        gr.Markdown("The next generation of AI-driven development.")
+        gr.Markdown("The flagship coding AI for the ZewpolOS ecosystem.")
 
     with gr.Row():
         # --- LEFT SIDEBAR ---
@@ -66,17 +72,18 @@ with gr.Blocks() as demo:
 
         # --- RIGHT MAIN ---
         with gr.Column(scale=4):
+            # Removed 'type="messages"' to fix the TypeError
             gr.ChatInterface(
                 fn=chat_engine,
                 additional_inputs=[model_dropdown],
-                type="messages",
                 examples=["Build a Python web scraper", "Explain ZewML logic"],
             )
 
-# Move the styling here to make Gradio 6.0 happy
+# Apply styling during launch
 if __name__ == "__main__":
     demo.launch(
         css=custom_css, 
         theme=gr.themes.Default(primary_hue="blue", secondary_hue="slate")
     )
 
+