Update app.py

app.py

@@ -21,6 +21,7 @@ from flask_cors import CORS, cross_origin
 # from flask_swagger_ui import get_swaggerui_blueprint
 import uuid
 import time
+from PIL import Image
 
 start_time = time.time()
 
@@ -219,6 +220,18 @@ def translate_text(text_prompt, target_language):
     )
     return response
 
+def chat_avatar(text_prompt):
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[{"role": "system", "content": "You are an interactive, conversational and helpful chatbot. Your role is to assist users by providing clear, engaging, and relevant responses based on their queries."},
+            {"role": "user", "content": f"Hi! I need help with something. Can you assist me with the following: {text_prompt}"},
+        ],
+        max_tokens = len(text_prompt) + 300 # Use the length of the input text
+        # temperature=0.3,
+        # stop=["Translate:", "Text:"]
+    )
+    return response
+
 
 
 @app.route("/run", methods=['POST'])
@@ -228,7 +241,9 @@ def generate_video():
     print('request:',request.method)
     try:
         if request.method == 'POST':
-            source_image = request.files['source_image']
+            # source_image = request.files['source_image']
+            image_path = '/home/user/app/images/avatar.jpg'
+            source_image = Image.open(image_path)
             text_prompt = request.form['text_prompt']
             print('Input text prompt: ',text_prompt)
             voice_cloning = request.form.get('voice_cloning', 'no')
@@ -245,11 +260,13 @@ def generate_video():
             print('preprocess selected: ',preprocess)
             ref_pose_video = request.files.get('ref_pose', None)
     
-            if target_language != 'original_text':
-                response = translate_text(text_prompt, target_language)
-                # response = await translate_text_async(text_prompt, target_language)
-                text_prompt = response.choices[0].message.content.strip()
-    
+            # if target_language != 'original_text':
+            #     response = translate_text(text_prompt, target_language)
+            #     # 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)