Update app.py

app.py

@@ -30,7 +30,11 @@ def handle_question(image, audio):
             #f"</audio>"
         )
     elif audio:  # Handle follow-up with just audio
-        message, answer_audio = process_inputs(session, audio_path=audio)
+
+        if not session.current_image:
+            return "No initial image found for the follow-up question.", None
+            
+        message, answer_audio = process_inputs(session, image=session.current_image, audio_path=audio)
 
         # Save the current image if it exists
         original_path = "uploaded_image.png"