Update app.py

app.py

@@ -35,12 +35,16 @@ if st.button("Analyze"):
         
         # Display JSON response
         st.json(response)
-        
-        # Additional visual representation & analysis based on API response.
-        # You will need to adjust this part based on the actual structure of the API response.
+
         try:
-            M = response["analyze"][0]["M"]
-            T = response["analyze"][0]["T"]
+            # Check if `response["analyze"]` is a string and parse it if true
+            analyze_data = response["analyze"]
+            if isinstance(analyze_data, str):
+                analyze_data = json.loads(analyze_data)
+
+            # Extract M and T values
+            M = analyze_data[0]["M"]
+            T = analyze_data[0]["T"]
 
             # Display traffic light system
             if M == 2 and T == 2:
@@ -49,8 +53,8 @@ if st.button("Analyze"):
                 st.write("🟡 Content is partly relevant/helpful")
             else:
                 st.write("🔴 Content is not relevant")
-        except KeyError:
-            st.error("Invalid API response format. Unable to extract M and T values.")
+        except (KeyError, IndexError, ValueError) as e:
+            st.error(f"Error extracting analysis results: {str(e)}")
+            st.error("Please check the API response format and adapt the code accordingly.")
     else:
-        st.warning("Please provide all inputs!")
-
+        st.warning("Please provide all inputs!")