Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import requests
+import json
+import re
 
 API_URL = "https://jj78yliqd7.execute-api.eu-central-1.amazonaws.com/dev/chatbot"
 
@@ -13,6 +15,14 @@ def get_response(prompt, history):
     else:
         return "Sorry, something went wrong. Please try again."
 
+# Regular expression to extract JSON object from text
+def extract_json(text):
+    json_data = re.search(r'\{.*\}', text, re.DOTALL)
+    if json_data:
+        return json_data.group(0)
+    else:
+        return None
+
 # Streamlit App
 st.sidebar.title("Wissensguru - IDTA")
 
@@ -39,9 +49,19 @@ if user_query is not None and user_query != "":
     # Generate bot's response
     response = get_response(user_query, st.session_state.chat_history)
     
-    # Display bot's response
-    with st.chat_message("AI"):
-        st.write(response)
+    # Extract JSON from bot's response
+    extracted_json = extract_json(response)
+    
+    if extracted_json:
+        # Parse JSON string into a dictionary
+        data_dict = json.loads(extracted_json)
+        
+        # Display JSON in Streamlit
+        st.json(data_dict)
+    else:
+        # Display bot's response if JSON extraction fails
+        with st.chat_message("AI"):
+            st.write(response)
     
     # Append messages to chat history
     st.session_state.chat_history.append(HumanMessage(content=user_query))