Update app.py

app.py

@@ -91,16 +91,40 @@ def handle_submission():
     #response = agent.run(message, tools=selected_tools)
     response = agent.run(message)
 
+    print(response)
+
     # Display the agent's response
-    # Display the agent's response
-    if response.startswith("Image:"):
-        # Display the image response
-        image_data = base64.b64decode(response.split(",")[1])
-        img = Image.open(io.BytesIO(image_data))
-        st.image(img)
-    else:
+    if isinstance(response, str):
         # Display the text response
         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
+    else:
+        # Handle unrecognized response type
+        st.warning("Unrecognized response type.")
+
+
+    # Display the agent's response
+    # Display the agent's response
+    #if response.startswith("Image:"):
+    #    # Display the image response
+    #    image_data = base64.b64decode(response.split(",")[1])
+    #    img = Image.open(io.BytesIO(image_data))
+    #    st.image(img)
+    #else:
+    #    # Display the text response
+    #    st.write(response)
 
 # Add a button to trigger the agent to respond again
 #st.button("Ask Again