Update app.py

app.py

@@ -156,19 +156,33 @@ def create_ui():
 # Main interface function
 @spaces.GPU()
 def main_interface(user_prompt, image=None, audio=None, voice_only=False, websearch=False):
+    print("Starting main_interface function")
     vqa_model.to(device='cuda', dtype=torch.bfloat16)
     tts_model.to("cuda")
     unet.to("cuda")
     image_pipe.to("cuda")
 
-    response = handle_input(user_prompt, image=image, audio=audio, websearch=websearch)
+    print(f"user_prompt: {user_prompt}, image: {image}, audio: {audio}, voice_only: {voice_only}, websearch: {websearch}")
+    
+    try:
+        response = handle_input(user_prompt, image=image, audio=audio, websearch=websearch)
+        print("handle_input function executed successfully")
+    except Exception as e:
+        print(f"Error in handle_input: {e}")
+        response = "Error occurred during processing."
 
     if voice_only:
-        audio_output = play_voice_output(response)
-        return "Response generated.", audio_output
+        try:
+            audio_output = play_voice_output(response)
+            print("play_voice_output function executed successfully")
+            return "Response generated.", audio_output
+        except Exception as e:
+            print(f"Error in play_voice_output: {e}")
+            return "Error occurred during voice output.", None
     else:
         return response, None
 
+
 # Launch the UI
 demo = create_ui()
 demo.launch()