Update app.py

app.py

@@ -120,16 +120,20 @@ def handle_submission():
     # ...
     
     # Display the agent's response
-    if "image" in response:
+    if isinstance(response, Image.Image):
         # Display the image response
-        image_data = base64.b64decode(response.split(",")[1])
-        img = Image.open(io.BytesIO(image_data))
-        st.image(img)
+        st.image(response)
     elif "audio" in response:
         # Handle audio response (replace with your audio rendering code)
         audio_data = base64.b64decode(response.split(",")[1])
         audio = AudioSegment.from_file(io.BytesIO(audio_data))
         st.audio(audio)
+    elif isinstance(response, AudioSegment):
+        # Handle audio response (replace with your audio rendering code)
+        st.audio(response)
+    elif isinstance(response, str):
+        # Display the text response
+        st.write(response)
     elif "text" in response:
         # Display the text response
         st.write(response)