Update app.py

app.py

@@ -272,7 +272,7 @@ def translate_text(text_prompt, target_language):
     )
     return response
 
-def chat_avatar(text_prompt):
+def openai_chat_avatar(text_prompt):
     response = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[{"role": "system", "content": "Answer using the minimum words you can ever use."},
@@ -284,6 +284,35 @@ def chat_avatar(text_prompt):
     )
     return response
 
+def ryzedb_chat_avatar(question):
+    url = "https://inference.dev.ryzeai.ai/chat/stream"
+    question = question + ". Summarize and Answer using the minimum words you can ever use."
+    payload = json.dumps({
+    "input": {
+    "chat_history": [],
+    "app_id": "af6b2bc719a14478adcff1d71c19dc00",
+    "question": question
+    },
+    "config": {}
+    })
+    headers = {
+        'Content-Type': 'application/json'
+    }
+    
+    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
+    
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
+
 
 
 @app.route("/run", methods=['POST'])
@@ -306,6 +335,7 @@ def generate_video():
                 return jsonify({'error': 'Input text prompt cannot be blank'}), 400
                 
             voice_cloning = request.form.get('voice_cloning', 'no')
+            chat_model_used = request.form.get('chat_model_used', 'openai')
             target_language = request.form.get('target_language', 'original_text')
             print('target_language',target_language)
             pose_style = int(request.form.get('pose_style', 1))
@@ -324,10 +354,30 @@ def generate_video():
             #     # response = await translate_text_async(text_prompt, target_language)
             #     text_prompt = response.choices[0].message.content.strip()
 
-            response = chat_avatar(text_prompt)
-            text_prompt = response.choices[0].message.content.strip()
-            app.config['text_prompt'] = text_prompt
-            print('Final text prompt: ',text_prompt)
+            if chat_model_used == 'ryzedb':
+                response = ryzedb_chat_avatar(text_prompt)
+                events  = response.split('\r\n\r\n')
+                content = None
+                for event in events:
+                # Split each event block by "\r\n" to get the lines
+                lines = event.split('\r\n')
+                if len(lines) > 1 and lines[0] == 'event: data':
+                    # Extract the JSON part from the second line and parse it
+                    json_data = lines[1].replace('data: ', '')
+                    try:
+                        data = json.loads(json_data)
+                        text_prompt = data.get('content')
+                        app.config['text_prompt'] = text_prompt
+                        print('Final output text prompt using ryzedb: ',text_prompt)
+                        break  # Exit the loop once content is found
+                    except json.JSONDecodeError:
+                        continue
+
+            else:
+                response = openai_chat_avatar(text_prompt)
+                text_prompt = response.choices[0].message.content.strip()
+                app.config['text_prompt'] = text_prompt
+                print('Final output text prompt using openai: ',text_prompt)
     
             source_image_path = save_uploaded_file(source_image, 'source_image.png',TEMP_DIR)
             print(source_image_path)