Update app.py

app.py

@@ -1,22 +1,25 @@
 import gradio as gr
 import json
 
-# Your existing interface
+# Function to query the model
 def query_model(prompt):
-    # This will use the featherless-ai provider
-    return prompt  # The actual model runs through gr.load
-
-# Create a JSON API endpoint
-def api_endpoint(prompt):
-    response = query_model(prompt)
-    return json.dumps({
+    # Simulate the response (replace with actual model logic)
+    response = f"Python function for {prompt}"
+    
+    # Format for Chatbot component
+    chatbot_messages = [
+        ["user", prompt],
+        ["assistant", response]
+    ]
+    
+    return chatbot_messages, json.dumps({
         "status": "success",
         "prompt": prompt,
         "response": response,
         "model": "open-r1/OlympicCoder-7B"
     })
 
-# Modified version with API capability
+# Create the Gradio interface
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=3):
@@ -40,7 +43,7 @@ with gr.Blocks() as demo:
     clear.click(lambda: None, None, chatbot, queue=False)
     
     # API endpoint
-    api_button.click(fn=api_endpoint, inputs=[api_input], outputs=[api_output])
+    api_button.click(fn=lambda x: query_model(x)[1], inputs=[api_input], outputs=[api_output])
 
 # Launch with debug info
 demo.launch(