Update app.py

app.py

@@ -12,9 +12,13 @@ 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,  # Use a random number for thread_id
+        "thread_id": thread_id,
         "nameM": model_name,
         "max_new_tokens": 4096,
         "top_p": 0.95,
@@ -27,18 +31,24 @@ 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": chat_history  # Include chat history
+        "chat_history": formatted_chat_history  # Ensure chat_history is properly formatted
     }
+    
     response = requests.post(url, headers=headers, json=data)
     
-    # Log the full response for debugging
-    print("API Response:", response.json())
+    # Log the status code and content for debugging
+    print(f"Status Code: {response.status_code}")
+    print(f"Response Content: {response.content}")
     
-    try:
-        return response.json()["message"]
-    except KeyError:
-        print("Key 'message' not found in the response")
-        return "Error: Unable to retrieve the message"
+    # 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}"
 
 # Gradio interface
 with gr.Blocks() as demo: