Update app.py

app.py

@@ -96,22 +96,19 @@ def handle_submission():
     # Display the agent's response
     if isinstance(response, str):
         # Display the text response
+        print("text")
         st.write(response)
-    elif isinstance(response, dict):
-        # Check for image, audio, or other types dynamically
-        response_keys = response.keys()
-        
-        if "Image" in response_keys:
-            # Display the image response
-            image_data = base64.b64decode(response["Image"].split(",")[1])
-            img = Image.open(io.BytesIO(image_data))
-            st.image(img)
-        elif "Audio" in response_keys:
-            # Handle audio response (replace with your audio rendering code)
-            st.audio(response["Audio"])
-        # Add more conditions for other types if needed
+    elif isinstance(response, Image):
+        # Display the image response
+        print("image")
+        st.image(response)
+    elif isinstance(response, Audio):
+        print("audio")
+        # Handle audio response (replace with your audio rendering code)
+        st.audio(response)
     else:
         # Handle unrecognized response type
+        print("warning")
         st.warning("Unrecognized response type.")