Update app.py

app.py

@@ -12,10 +12,6 @@ def api_call(prompt, thread_id, model_name, chat_history):
         "key": api_key,
         "Content-Type": "application/json"
     }
-    
-    # Ensure chat_history is formatted as a list of strings
-    formatted_chat_history = [[str(item[0]), str(item[1])] if item[1] is not None else [str(item[0]), ""] for item in chat_history]
-    
     data = {
         "prompt": prompt,
         "thread_id": thread_id,
@@ -31,48 +27,41 @@ def api_call(prompt, thread_id, model_name, chat_history):
         "persist_data": True,
         "max_chat_history_tokens": 2000,
         "max_chat_history_days": 10,
-        "chat_history": formatted_chat_history  # Ensure chat_history is properly formatted
+        "chat_history": chat_history  # Include chat history
     }
-    
     response = requests.post(url, headers=headers, json=data)
-    
-    # Log the status code and content for debugging
-    print(f"Status Code: {response.status_code}")
-    print(f"Response Content: {response.content}")
-    
-    # Check if the response contains JSON data
-    if response.status_code == 200:
-        try:
-            return response.json().get("message", "Error: 'message' key not found in the response")
-        except ValueError:
-            print("Error parsing JSON response")
-            return "Error: Unable to parse the response"
-    else:
-        return f"Error: Received status code {response.status_code}"
+    try:
+        return response.json()["message"]
+    except requests.exceptions.JSONDecodeError:
+        print("Error: Invalid JSON response")
+        return "Error: Invalid JSON response"
+
+# Initialize the chat history
+chat_history = []
 
 # Gradio interface
 with gr.Blocks() as demo:
     model_name = gr.Dropdown(choices=["OpenAI", "Claude"], label="Select Model")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Your Message")
-    state = gr.State([])  # Initialize state for chat history
 
-    def respond(message, model_name, chat_history):
-        # Append the user's message to the chat history immediately
-        chat_history.append((message, None))
-        # Get the bot's response
+    def respond(message, model_name):
+        global chat_history
+        # Make the API call with the current message
         bot_message = api_call(message, random.randint(1, 10000), model_name, chat_history)
-        # Update the last message with the bot's response
-        chat_history[-1] = (message, bot_message)
+        # Append the current message and response to chat history
+        chat_history.append((message, bot_message))
         return "", chat_history
 
     def reset_chat_history(model_name):
         # Reset the chat history when the model_name changes
-        return [], []
+        global chat_history
+        chat_history = []
+        return [], chat_history
 
     # Link the reset function to the model_name dropdown
-    model_name.change(fn=reset_chat_history, inputs=model_name, outputs=[chatbot, state])
+    model_name.change(fn=reset_chat_history, inputs=model_name, outputs=[chatbot, gr.State(chat_history)])
 
-    msg.submit(fn=respond, inputs=[msg, model_name, state], outputs=[msg, chatbot])
+    msg.submit(fn=respond, inputs=[msg, model_name], outputs=[msg, chatbot])
 
-demo.launch(share=True)
+demo.launch()