Update app.py

app.py

@@ -289,10 +289,15 @@ def generate_video():
             print('preprocess selected: ',preprocess)
             ref_pose_video = request.files.get('ref_pose', None)
     
-            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)
+            if chat_model_used == 'openai':
+                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)
+
+            else:
+                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)