Update app.py

app.py

@@ -302,18 +302,6 @@ def save_uploaded_file(file, filename,TEMP_DIR):
 
 client = OpenAI(api_key="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")
 
-def translate_text(text_prompt, target_language):
-    response = client.chat.completions.create(
-        model="gpt-4o-mini",
-        messages=[{"role": "system", "content": "You are a helpful language translator assistant."},
-            {"role": "user", "content": f"Translate completely without hallucination, end to end and the ouput should just be the translation of the text prompt and nothing else, and give the following text to {target_language} language and the text is: {text_prompt}"},
-        ],
-        max_tokens = len(text_prompt) + 200 # Use the length of the input text
-        # temperature=0.3,
-        # stop=["Translate:", "Text:"]
-    )
-    return response
-
 def openai_chat_avatar(text_prompt):
     response = client.chat.completions.create(
         model="gpt-4o-mini",
@@ -392,7 +380,7 @@ def generate_video():
                 
             voice_cloning = request.form.get('voice_cloning', 'no')
             image_hardcoded = request.form.get('image_hardcoded', 'yes')
-            chat_model_used = request.form.get('chat_model_used', 'openai')
+            chat_model_used = request.form.get('chat_model_used', 'ryzedb')
             target_language = request.form.get('target_language', 'original_text')
             print('target_language',target_language)
             pose_style = int(request.form.get('pose_style', 1))
@@ -405,11 +393,7 @@ def generate_video():
             preprocess = request.form.get('preprocess', 'crop')
             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 chat_model_used == 'ryzedb':
                 response = ryzedb_chat_avatar(text_prompt)
@@ -431,16 +415,11 @@ def generate_video():
                             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)
+                print("No Ryze database found")
     
             source_image_path = save_uploaded_file(source_image, 'source_image.png',TEMP_DIR)
             print(source_image_path)
-    
-            # driven_audio_path = await voice_cloning_async(voice_cloning, voice_gender, text_prompt, user_voice)
-    
+        
             if voice_cloning == 'no':
                 if voice_gender == 'male':
                     voice = 'echo'