integrating chartgpt

README.md

@@ -10,3 +10,9 @@ pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+Models Used:
+- https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1
+- https://huggingface.co/meta-llama/Llama-4-Maverick-17B-128E-Instruct
+- https://huggingface.co/MiniMaxAI/MiniMax-M1-80k

app.py

@@ -1,26 +1,26 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+from transformers import pipeline
 
-# Setup clients for each provider
+# === Inference Clients ===
 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"])
 
+# === ChartGPT pipeline ===
+chart_pipe = pipeline("text2text-generation", model="yuan-tian/chartgpt-llama3")
+
+# === Chat Handler ===
 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)":
             content = [{"type": "text", "text": prompt}]
             if image_url:
-                content.append({
-                    "type": "image_url",
-                    "image_url": {"url": image_url}
-                })
-
+                content.append({"type": "image_url", "image_url": {"url": image_url}})
             messages = [{"role": "user", "content": content}]
             completion = llama_client.chat.completions.create(
                 model="meta-llama/Llama-4-Maverick-17B-128E-Instruct",
@@ -28,7 +28,6 @@ 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(
@@ -37,7 +36,6 @@ 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(
@@ -48,34 +46,42 @@ def chat_with_model(model_choice, prompt, image_url):
 
         else:
             return "Unsupported model selected."
-
     except Exception as e:
         return f"Error: {e}"
 
-# Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Unified Chatbot Interface")
-    gr.Markdown("Supports LLaMA 4 (with optional image), MiniMax, and Mistral.")
+# === ChartGPT Handler ===
+def generate_chart_code(prompt):
+    try:
+        response = chart_pipe(prompt, max_new_tokens=512)[0]["generated_text"]
+        return response
+    except Exception as e:
+        return f"ChartGPT error: {e}"
 
-    model_dropdown = gr.Dropdown(
-        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")
+# === Gradio UI ===
+with gr.Blocks() as demo:
+    gr.Markdown("## 🔥 Multi-Tool AI Space: Chat + Chart Generator")
 
-    submit_btn = gr.Button("Generate Response")
-    output_box = gr.Textbox(label="Response", lines=8)
+    with gr.Tabs():
+        with gr.Tab("💬 Multimodel Chat"):
+            model_dropdown = gr.Dropdown(
+                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)
+            submit_btn.click(chat_with_model, [model_dropdown, prompt_input, image_url_input], output_box)
 
-    submit_btn.click(
-        fn=chat_with_model,
-        inputs=[model_dropdown, prompt_input, image_url_input],
-        outputs=output_box
-    )
+        with gr.Tab("📊 Chart Generator (ChartGPT)"):
+            chart_prompt = gr.Textbox(label="Enter data analysis prompt", placeholder="Generate a bar chart comparing 2023 sales in North America and Europe", lines=3)
+            chart_btn = gr.Button("Generate Chart Code")
+            chart_output = gr.Textbox(label="ChartGPT Code Output", lines=16)
+            chart_btn.click(generate_chart_code, chart_prompt, chart_output)
 
 demo.launch()