Update app.py

app.py

@@ -1,21 +1,24 @@
 import gradio as gr
-from litellm import completion
+from openai import OpenAI
 import os
 
 # Puter API Konfiguration
 PUTER_TOKEN = os.getenv("PUTER_API_KEY")
 PUTER_API_BASE = "https://api.puter.com/puterai/openai/v1/"
 
+client = OpenAI(
+    api_key=PUTER_TOKEN,
+    base_url=PUTER_API_BASE,
+)
+
 def respond(message, history, model):
     if history is None:
         history = []
     messages = [{"role": "user", "content": message}]
     try:
-        response = completion(
-            model=f"openai/{model}",
+        response = client.chat.completions.create(
+            model=model,
             messages=messages,
-            api_key=PUTER_TOKEN,
-            api_base=PUTER_API_BASE,
         )
         assistant_reply = response.choices[0].message.content
     except Exception as e:
@@ -66,34 +69,25 @@ body, .gradio-container {
 }
 """
 
-# ✅ css passed to gr.Blocks, not demo.launch()
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("# puter.js in Python via LiteLLM", elem_id="title")
-
-    # ✅ state declared INSIDE gr.Blocks before it's referenced
+    gr.Markdown("# puter.js in Python via OpenAI SDK", elem_id="title")
     state = gr.State([])
-
     with gr.Row():
         output = gr.Markdown(value="", elem_id="output-markdown")
-
     with gr.Row():
         model = gr.Dropdown(
             choices=["gpt-4o-mini", "gpt-5.2-codex", "claude-sonnet-4", "gemini-2.5-flash"],
             value="gpt-4o-mini",
             label="Modell auswählen"
         )
-
     with gr.Row():
         msg = gr.Textbox(
             label="Deine Nachricht",
             placeholder="Schreibe hier...",
             lines=2,
         )
-
     with gr.Row():
         submit_btn = gr.Button("Senden")
-
-    # ✅ state now exists when referenced here
     submit_btn.click(
         fn=respond,
         inputs=[msg, state, model],