Update app.py

app.py

@@ -712,41 +712,10 @@ system_prompt = f"""
 {instruction}
 """
 
-client = openai.OpenAI(api_key=OPENAI_API)
-
-class MyOpenAIChatModel:
-    def __init__(self, model_name: str = "gpt-4.1-mini"):
-        self.model_name = model_name
-
-    def __call__(self, messages, stop_sequences=None, **kwargs):
-        """
-        Adapter so smolagents' CodeAgent can call the OpenAI client.
-
-        CodeAgent passes `messages` as a list of either:
-        - dicts with 'role' and 'content'
-        - or Message-like objects with .role and .content
-        """
-        normalized = []
-        for m in messages:
-            if isinstance(m, dict):
-                normalized.append(m)
-            else:
-                role = getattr(m, "role", None)
-                content = getattr(m, "content", None)
-                if role is None or content is None:
-                    raise ValueError(
-                        f"Unexpected message type: {type(m)}; cannot extract role/content"
-                    )
-                normalized.append({"role": role, "content": content})
-
-        resp = client.chat.completions.create(
-            model=self.model_name,
-            messages=normalized,
-            stop=stop_sequences,
-        )
-        return resp.choices[0].message.content
-
-model = MyOpenAIChatModel("gpt-4.1-mini")
+model = smolagents.OpenAIServerModel(
+    model_id="gpt-4.1-mini",
+    api_key=OPENAI_API,
+)
 
 agent = smolagents.CodeAgent(
     tools=[
@@ -766,44 +735,53 @@ agent = smolagents.CodeAgent(
     max_steps=10,
     verbosity_level=2,  # show steps in logs for class demo
 )
-
-def respond(message, history):
-    try:
-        out = str(agent.run(message))
-    except Exception as e:
-        out = f"[Error] {e}"
-
-    history = (history or []) + [(message, out)]
-    return "", history
     
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Ocean()) as demo:
+    
+    # Centered title and description using HTML
     gr.HTML("""
-        <div style="text-align: center; font-family: 'Arial', sans-serif;">
-            <h1 style="color:#1f77b4; margin-bottom: 20px; font-weight: 300;">
-                🤖 TechSpark AI Assistant
-            </h1>
-            <p style="margin-top: 0; font-weight: 300; font-size: 16px; color:#555;">
-                Welcome to the TechSpark AI Assistant!<br>
-                Ask anything about TechSpark staff, tools, courses or location of tools.<br>
-                This assistant is powered by OpenAI's GPT model via smolagents, 
-                accessing accurate information from our curated dataset verified by TechSpark staff!
-            </p>
-        </div>
-    """)
+            <div style="text-align: center; font-family: 'Arial', sans-serif;">
+                <h1 style="color:#1f77b4; margin-bottom: 20px; font-weight: 300;">
+                    🤖 TechSpark AI Assistant
+                </h1>
+                
+                <p style="margin-top: 0; font-weight: 300; font-size: 16px; color:#555;">
+                    Welcome to the TechSpark AI Assistant!<br>
+                    Ask anything about TechSpark staff, tools, courses or location of tools.<br>
+                    This assistant is powered by OpenAI's GPT model via smolagents, 
+                    accessing accurate information from our curated dataset verified by TechSpark staff!
+                </p>
+            </div>
+            """)
 
     chat = gr.Chatbot(height=420)
     inp = gr.Textbox(placeholder="Ask your question in natural language.", label="Your question")
 
-    inp.submit(respond, [inp, chat], [inp, chat])
+    # No gr.State for agent — just close over `agent`
+    def respond(message, history):
+        try:
+            # 1. Use agent.chat() to maintain internal history
+            out = str(agent.run(message))
+        except Exception as e:
+            out = f"[Error] {e}"
+
+        # This just updates the Gradio UI history
+        history = (history or []) + [(message, out)]
+        return "", history
 
     gr.Examples(
         examples=[
             "Who is Ed?",
             "Who to talk to to create a wooden table?",
-            "How can I access the laser cutters?",
-            "How can I go to the welding space?"
+            "How to access the laser cutters?",
+            "How to go to the welding space?",
         ],
         inputs=[inp],
+        outputs=[inp, chat],
+        fn=respond,
+        cache_examples=False, # Set to False for dynamic content or to avoid caching issues
     )
 
+    inp.submit(respond, [inp, chat], [inp, chat])
+
 demo.launch()