Update app.py

app.py

@@ -79,7 +79,12 @@ def stream_response(message, history, character):
     try:
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()
-        content = response.json()["choices"][0]["message"]["content"]
+        data = response.json()
+        if "choices" not in data:
+            # Yield the full response data for debugging.
+            yield f"Error: API returned an unexpected response: {data}"
+            return
+        content = data["choices"][0]["message"]["content"]
         
         stream_response = ""
         # Simulate streaming by yielding token-by-token.
@@ -93,21 +98,19 @@ def stream_response(message, history, character):
 def chat(user_message, history, character):
     """
     Appends the user message to the conversation history, then streams the assistant's reply.
-    The conversation history is a list of dictionaries with keys "role" and "content".
     """
     # Ensure history is a list.
     history = history or []
     history = history.copy()
-    # Append user message.
+    # Append the user's message.
     history.append({"role": "user", "content": user_message})
     
-    # Create a generator for the streaming response.
     full_response = ""
     for partial in stream_response(user_message, history, character):
         full_response = partial
         # Yield the conversation updated with the current assistant response.
         yield history + [{"role": "assistant", "content": full_response}]
-    # Once done, append the final assistant message.
+    # Append the final assistant message.
     history.append({"role": "assistant", "content": full_response})
     return history
 
@@ -148,10 +151,12 @@ with gr.Blocks(css=css) as demo:
     # State to hold conversation history.
     state = gr.State([])
 
-    # When user submits text (via button or Enter), update chat.
-    msg.submit(fn=lambda user_message, history, choice: chat(user_message, history, clean_choice(choice)),
-               inputs=[msg, state, model_dropdown],
-               outputs=[chatbot, state],
-               show_progress=True)
+    # When user submits text, update chat.
+    msg.submit(
+        fn=lambda user_message, history, choice: chat(user_message, history, clean_choice(choice)),
+        inputs=[msg, state, model_dropdown],
+        outputs=[chatbot, state],
+        show_progress=True
+    )
     
     demo.launch(share=True)