Update app.py

app.py

@@ -8,6 +8,7 @@ from project_model import process_inputs, session
 def handle_question(image, audio):
     try:
         if image and audio:  # Handle the initial question with image and audio
+            session.current_image = image
             message, answer_audio = process_inputs(session, image=image, audio_path=audio)
     
             # Save images
@@ -33,7 +34,7 @@ def handle_question(image, audio):
                 #f"</audio>"
             )
         elif audio:  # Handle follow-up with just audio
-    
+            
             if not session.current_image:
                 return "No initial image found for the follow-up question.", None