added mistral model

app.py

@@ -2,17 +2,18 @@ import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Clients for both providers
+# Setup clients for each provider
 llama_client = InferenceClient(provider="sambanova", api_key=os.environ["HF_TOKEN"])
 minimax_client = InferenceClient(provider="novita", api_key=os.environ["HF_TOKEN"])
+mistral_client = InferenceClient(provider="together", api_key=os.environ["HF_TOKEN"])
 
 def chat_with_model(model_choice, prompt, image_url):
     if not prompt:
         return "Please enter a text prompt."
 
     try:
+        # LLaMA 4 supports optional image input
         if model_choice == "LLaMA 4 (SambaNova)":
-            # Prepare message with optional image
             content = [{"type": "text", "text": prompt}]
             if image_url:
                 content.append({
@@ -27,6 +28,7 @@ def chat_with_model(model_choice, prompt, image_url):
             )
             return completion.choices[0].message.content
 
+        # MiniMax: Text only
         elif model_choice == "MiniMax M1 (Novita)":
             messages = [{"role": "user", "content": prompt}]
             completion = minimax_client.chat.completions.create(
@@ -35,6 +37,15 @@ def chat_with_model(model_choice, prompt, image_url):
             )
             return completion.choices[0].message.content
 
+        # Mistral: Text only
+        elif model_choice == "Mistral Mixtral-8x7B (Together)":
+            messages = [{"role": "user", "content": prompt}]
+            completion = mistral_client.chat.completions.create(
+                model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+                messages=messages
+            )
+            return completion.choices[0].message.content
+
         else:
             return "Unsupported model selected."
 
@@ -43,17 +54,21 @@ def chat_with_model(model_choice, prompt, image_url):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Multimodel Chatbot: LLaMA 4 & MiniMax M1")
-    gr.Markdown("Choose a model, enter your prompt, and optionally add an image URL for LLaMA.")
+    gr.Markdown("## 🤖 Unified Chatbot Interface")
+    gr.Markdown("Supports LLaMA 4 (with optional image), MiniMax, and Mistral.")
 
     model_dropdown = gr.Dropdown(
-        choices=["LLaMA 4 (SambaNova)", "MiniMax M1 (Novita)"],
+        choices=[
+            "LLaMA 4 (SambaNova)",
+            "MiniMax M1 (Novita)",
+            "Mistral Mixtral-8x7B (Together)"
+        ],
         value="LLaMA 4 (SambaNova)",
         label="Select Model"
     )
     prompt_input = gr.Textbox(label="Text Prompt", placeholder="Ask something...", lines=2)
     image_url_input = gr.Textbox(label="Optional Image URL (for LLaMA only)", placeholder="https://example.com/image.jpg")
-    
+
     submit_btn = gr.Button("Generate Response")
     output_box = gr.Textbox(label="Response", lines=8)