Update app.py

app.py

@@ -191,12 +191,20 @@ def chat_function_gpu(message, history):
         # Decode and return response
         decoded_output = tokenizer.decode(output[len(tokenized["input_ids"][0]) :])
         
-        # Return in tuple format for Gradio chatbot
-        return history + [[message, decoded_output]]
+        # Return in correct Gradio chatbot format
+        new_history = history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": decoded_output}
+        ]
+        return new_history
 
     except Exception as e:
         error_msg = f"Error processing your request: {str(e)}"
-        return history + [[message, error_msg]]
+        new_history = history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": error_msg}
+        ]
+        return new_history
 
 # Fallback CPU function for when GPU is not available
 def chat_function_cpu(message, history):
@@ -241,12 +249,20 @@ def chat_function_cpu(message, history):
         # Decode and return response
         decoded_output = tokenizer.decode(output[len(tokenized["input_ids"][0]) :])
         
-        # Return in tuple format for Gradio chatbot
-        return history + [[message, decoded_output]]
+        # Return in correct Gradio chatbot format
+        new_history = history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": decoded_output}
+        ]
+        return new_history
 
     except Exception as e:
         error_msg = f"Error processing your request: {str(e)}"
-        return history + [[message, error_msg]]
+        new_history = history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": error_msg}
+        ]
+        return new_history
 
 # Create custom theme optimized for ZeroGPU
 custom_theme = gr.themes.Soft(
@@ -264,7 +280,7 @@ custom_theme = gr.themes.Soft(
 )
 
 # Create Gradio interface with ZeroGPU support - Gradio 6 syntax
-with gr.Blocks(fill_height=True) as demo:
+with gr.Blocks(fill_height=True, theme=custom_theme) as demo:
     gr.Markdown("""
     # 🚀 Mistral Vibe - AI Coding Assistant
     
@@ -275,7 +291,8 @@ with gr.Blocks(fill_height=True) as demo:
     
     chatbot = gr.Chatbot(
         height=600,
-        label="Chat with Mistral Vibe"
+        label="Chat with Mistral Vibe",
+        type="messages"  # Use messages format for Gradio 6
     )
     
     with gr.Row():
@@ -341,21 +358,6 @@ with gr.Blocks(fill_height=True) as demo:
 # Launch with custom theme and ZeroGPU settings - Gradio 6 syntax
 demo.queue()  # Enable queue separately in Gradio 6
 demo.launch(
-    theme=custom_theme,
-    footer_links=[
-        {
-            "label": "Built with anycoder",
-            "url": "https://huggingface.co/spaces/akhaliq/anycoder"
-        },
-        {
-            "label": "Mistral AI",
-            "url": "https://mistral.ai"
-        },
-        {
-            "label": "Hugging Face ZeroGPU",
-            "url": "https://huggingface.co/docs/hub/spaces-zerogpu"
-        }
-    ],
     share=False,
     max_threads=4,
     show_error=True