Update app.py

app.py

@@ -22,10 +22,10 @@ footer {
 select {
     background-color: #2a2a40;
     color: white;
-    padding: 10px;
+    padding: 6px 10px;
     border-radius: 8px;
     border: 1px solid #444;
-    width: 100%;
+    width: 300px;
 }
 option {
     background-color: #2a2a40;
@@ -41,7 +41,6 @@ HEADERS = {
     "Authorization": f"Bearer {ACCESS_TOKEN}",
 }
 
-# Define our system prompts
 PROMPTS = {
     "Elon Ma (Official)": (
         "You are Elon Ma, a Chinese car salesman selling the Edision Model S.\n"
@@ -79,12 +78,12 @@ def respond(message, history, character):
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
         content = response.json()["choices"][0]["message"]["content"]
-
+        
         stream_response = ""
         for token in content.split():
             stream_response += token + " "
             time.sleep(0.02)
-            # Yield a dictionary so that ChatInterface's streaming works correctly
+            # Yield a message dictionary for streaming updates.
             yield {"role": "assistant", "content": stream_response.strip()}
     except Exception as e:
         yield {"role": "assistant", "content": f"Error: {str(e)}"}
@@ -93,8 +92,6 @@ with gr.Blocks(css=css) as demo:
     # 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>")
     
-    # Create a dropdown with rich labels in the text.
-    # The dropdown options include emojis and descriptions.
     with gr.Row():
         with gr.Column():
             character = gr.Dropdown(
@@ -107,7 +104,6 @@ with gr.Blocks(css=css) as demo:
                 label="Model"
             )
     
-    # Helper function to extract the clean key for our prompts.
     def clean_choice(choice):
         if "Elon" in choice:
             return "Elon Ma (Official)"
@@ -117,11 +113,11 @@ with gr.Blocks(css=css) as demo:
             return "Mr. Shortreed (Official)"
         return "Elon Ma (Official)"
     
-    # ChatInterface with streaming enabled
     chatbot = gr.ChatInterface(
         lambda msg, hist, char: respond(msg, hist, clean_choice(char)),
         additional_inputs=[character],
-        type="messages"  # use openai-style message format
+        type="messages",   # Use OpenAI-style message format.
+        stream=True        # Enable streaming.
     )
     
     demo.launch(share=True)