Update app.py

app.py

@@ -1,30 +1,53 @@
 import gradio as gr
 from openai import OpenAI
 import os
+import time
 
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-def chat(message):
+def stream_chat(message, history):
+    history = history or []
+    history.append((message, ""))
+
     response = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
             {"role": "system", "content": "You are a helpful assistant."},
+            *[
+                {"role": "user" if i % 2 == 0 else "assistant", "content": m}
+                for pair in history[:-1]
+                for i, m in enumerate(pair)
+            ],
             {"role": "user", "content": message}
-        ]
+        ],
+        stream=True
     )
-    return response.choices[0].message.content
+
+    partial = ""
+    for chunk in response:
+        if chunk.choices[0].delta.get("content"):
+            partial += chunk.choices[0].delta["content"]
+            history[-1] = (message, partial)
+            yield history
 
 css = """
 body {
     background-color: #0f0f0f;
 }
 .gradio-container {
-    max-width: 700px !important;
+    max-width: 720px !important;
     margin: auto;
 }
+.message.user {
+    background: #343541 !important;
+}
+.message.bot {
+    background: #1e1e1e !important;
+}
 textarea {
     background-color: #1e1e1e !important;
     color: white !important;
+    border-radius: 12px !important;
 }
 """
 
@@ -32,25 +55,26 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         <h1 style="text-align:center;">💬 ChatGPT‑style AI</h1>
-        <p style="text-align:center; color:gray;">
-        Powered by OpenAI
+        <p style="text-align:center; color:#888;">
+        Powered by OpenAI ChatGPT
         </p>
         """
     )
 
-    chatbot = gr.Chatbot(height=400)
+    chatbot = gr.Chatbot(
+        height=420,
+        avatar_images=("https://i.imgur.com/7k12EPD.png", "https://i.imgur.com/9k6dKQy.png")
+    )
+
     msg = gr.Textbox(
-        placeholder="Send a message...",
-        show_label=False
+        placeholder="Send a message…",
+        show_label=False,
+        autofocus=True
     )
-    clear = gr.Button("Clear")
 
-    def respond(message, history):
-        reply = chat(message)
-        history.append((message, reply))
-        return "", history
+    clear = gr.Button("🗑 Clear chat")
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    msg.submit(stream_chat, [msg, chatbot], chatbot)
     clear.click(lambda: [], None, chatbot)
 
 demo.launch()