Update app.py

app.py

@@ -174,34 +174,38 @@ def openai_chat_avatar(text_prompt):
 
 def ryzedb_chat_avatar(question, app_id):
     url = "https://inference.dev.ryzeai.ai/v2/chat/stream"
-    # question = question + ". Summarize the answer in one line."
-    # print("question",question)
-    payload = json.dumps({
+    print("ryze db question",question)
+    # thread_id = str(random.randint(1000, 99999))
+    # print("Generated thread_id:", thread_id)
+    payload = {
     "input": {
-    "chat_history": [],
-    "app_id": app_id,
-    "question": question
+        "app_id": app_id,
+        "query": question,
+        "chat_history": []
     },
     "config": {
         "thread_id": "123456"
-    })
+    }}
+    
     headers = {
         'Content-Type': 'application/json'
     }
-    
+    response = requests.post(url, json=payload, headers=headers, stream=True)
     try:
-        # Send the POST request
-        response = requests.request("POST", url, headers=headers, data=payload)
-        
-        # Check for successful request
-        response.raise_for_status()
-        
-        # Return the response JSON
-        return response.text
+        raw_text = response.text.strip()
+    
+        if raw_text.startswith("data:"):
+            raw_text = raw_text[len("data:"):].strip()
     
-    except requests.exceptions.RequestException as e:
-        print(f"An error occurred: {e}")
-        return None
+        json_data = json.loads(raw_text)
+    
+        response  = json_data.get("content", "")
+        return response
+
+    except Exception as e:
+        print("Error parsing response:", e)
+        return ""
+
 
 
 def custom_cleanup(temp_dir, exclude_dir):