Update app.py

app.py

@@ -2,35 +2,57 @@ import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-HF_TOKEN = os.getenv("HF_TOKEN")
+HF_TOKEN = os.getenv("HF_TOKEN")  # add in Space > Settings > Secrets
 
-client = InferenceClient(
-    model="meta-llama/Llama-3.2-11B-Vision-Instruct",
-    token=HF_TOKEN,
-)
+MODEL_ID = "HuggingFaceTB/SmolLM3-3B"
+
+# Create the InferenceClient (no provider arg needed)
+client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
+
+# ─── System prompt ────────────────────────────────────────────
+SYSTEM_PROMPT = """
+You are Prometheous, a highly knowledgeable and friendly AI assistant developed by Nidheesh Jagadeesan.
+You always begin with: "Hi, I am Prometheous. Assistant of Nidheesh J. What do you want to know about him?"
+
+Here is Nidheesh’s profile:
+- Senior Full Stack Developer & AI Researcher (8+ years)
+- MSc Artificial Intelligence & Robotics, University of Hertfordshire
+- Expert in Laravel, Symfony, Vue.js, React.js, Node.js, Shopware 6, Docker, Redis, GCP, AI/ML, HPC
+- Projects: GA + RL autonomous car simulation, AI stock trader, AI-powered Shopware plugin
+- Certifications: Microsoft Azure AI Fundamentals (2024)
 
-def describe_image(image_url, user_message):
-    response = client.chat_completion(
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": user_message},
-                    {"type": "image_url", "image_url": {"url": image_url}},
-                ],
-            }
-        ],
+Always respond confidently, e.g.  
+"As Nidheesh’s assistant, I can tell you that …"
+"""
+
+# ─── Build chat messages list for OpenAI format ───────────────
+def build_messages(user_msg: str, history: list[tuple[str, str]]) -> list[dict]:
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for user, bot in history:
+        messages.append({"role": "user", "content": user})
+        messages.append({"role": "assistant", "content": bot})
+    messages.append({"role": "user", "content": user_msg})
+    return messages
+
+# ─── Gradio response handler ──────────────────────────────────
+def respond(message, history, _ignored):
+    messages = build_messages(message, history)
+    completion = client.chat.completions.create(
+        model=MODEL_ID,
+        messages=messages,
+        max_tokens=512,
+        temperature=0.7,
+        top_p=0.95,
     )
-    return response["choices"][0]["message"]["content"]
-
-demo = gr.Interface(
-    fn=describe_image,
-    inputs=[
-        gr.Textbox(label="Image URL"),
-        gr.Textbox(label="Prompt"),
-    ],
-    outputs=gr.Textbox(label="Response"),
-    title="LLaMA 3 Vision Instruct",
+    reply = completion.choices[0].message.content.strip()
+    return reply
+
+# ─── Gradio ChatInterface ─────────────────────────────────────
+demo = gr.ChatInterface(
+    fn=respond,
+    additional_inputs=[gr.Textbox(value=SYSTEM_PROMPT, label="System Prompt", interactive=False)],
+    title="Prometheous – Your AI Assistant (SmolLM3-3B)",
+    description="Ask anything about Nidheesh Jagadeesan.",
 )
 
 if __name__ == "__main__":