Update app.py

app.py

@@ -41,6 +41,7 @@ HEADERS = {
     "Authorization": f"Bearer {ACCESS_TOKEN}",
 }
 
+# Define system prompts for each model.
 PROMPTS = {
     "Elon Ma (Official)": (
         "You are Elon Ma, a Chinese car salesman selling the Edision Model S.\n"
@@ -83,17 +84,18 @@ def respond(message, history, character):
         for token in content.split():
             stream_response += token + " "
             time.sleep(0.02)
-            # Yield a message dictionary for streaming updates.
+            # Yield each update as a message dictionary.
             yield {"role": "assistant", "content": stream_response.strip()}
     except Exception as e:
         yield {"role": "assistant", "content": f"Error: {str(e)}"}
 
 with gr.Blocks(css=css) as demo:
-    # Header with QClone Public label
+    # Header with QClone Public label.
     gr.HTML("<h1>QClone <span style='background-color:#3b82f6;color:white;font-size:0.75em;padding:2px 6px;border-radius:5px;margin-left:8px;'>Public</span></h1>")
     
     with gr.Row():
         with gr.Column():
+            # Native dropdown with emojis and descriptions.
             character = gr.Dropdown(
                 choices=[
                     "Elon Ma (Official) 🟡 - Broken English salesman",
@@ -104,6 +106,7 @@ with gr.Blocks(css=css) as demo:
                 label="Model"
             )
     
+    # Helper to clean up the selected choice.
     def clean_choice(choice):
         if "Elon" in choice:
             return "Elon Ma (Official)"
@@ -113,11 +116,11 @@ with gr.Blocks(css=css) as demo:
             return "Mr. Shortreed (Official)"
         return "Elon Ma (Official)"
     
+    # ChatInterface using OpenAI-style messages.
     chatbot = gr.ChatInterface(
         lambda msg, hist, char: respond(msg, hist, clean_choice(char)),
         additional_inputs=[character],
-        type="messages",   # Use OpenAI-style message format.
-        stream=True        # Enable streaming.
+        type="messages"
     )
     
     demo.launch(share=True)