Update app.py

app.py

@@ -399,6 +399,7 @@ def generate_video():
 
 
             if chat_model_used == 'ryzedb':
+                start_time_ryze = time.time()
                 response = ryzedb_chat_avatar(text_prompt, app_id)
                 events  = response.split('\r\n\r\n')
                 content = None
@@ -412,7 +413,10 @@ def generate_video():
                             data = json.loads(json_data)
                             text_prompt = data.get('content')
                             app.config['text_prompt'] = text_prompt
+                            end_time_ryze = time.time()
+                            diff = end_time_ryze - start_time_ryze
                             print('Final output text prompt using ryzedb: ',text_prompt)
+                            print('Time to get response from ryzedb: ',diff)
                             break  # Exit the loop once content is found
                         except json.JSONDecodeError:
                             continue