Update app.py

app.py

@@ -67,30 +67,41 @@ if submit_button and text_input:
     api_key = os.environ.get('WL_KEY', 'default_key')  # Use a default or handle the case where the key is not set
     api_response = fact_check_statement(text_input, api_key)
 
-if "error" not in api_response:
-    # Access the fields directly from api_response
-    claim_reviewed = api_response.get("claimReviewed", "N/A")
-    review_rating = api_response.get("reviewRating", {})
-    rating_value = review_rating.get("ratingValue", "N/A")
-    alternate_name = review_rating.get("alternateName", "N/A")  # Extract the alternateName
-    review_body = api_response.get("reviewBody", "N/A")
+    if "error" not in api_response:
+        # Extract the 'response' string from api_response
+        response_str = api_response.get("response", "{}")
 
-    # Display the results
-    if rating_value.isdigit():
-        st.progress(int(rating_value) / 5, "Fact-checking in progress.")
-    else:
-        st.write("Rating Value: ", rating_value)
-    st.write(f"**Review Body**: {review_body}")
+        # Remove the triple backticks from the response string if they exist
+        response_str = response_str.strip('`')
 
-    # Create a two-column layout
-    col1, col2 = st.columns(2)
+        # Parse the string as JSON
+        try:
+            inner_json = json.loads(response_str)
 
-    # Use the metric widget to display the rating and alternate name
-    col1.metric("Rating", rating_value)
-    col2.metric("Verdict", alternate_name)
-    
-    #with st.expander("Here is the final JSON-LD"):
-    #    st.json(api_response)  # Display the entire JSON-LD data
+            # Now, extract the information from inner_json as before
+            claim_reviewed = inner_json.get("claimReviewed", "N/A")
+            review_rating = inner_json.get("reviewRating", {})
+            rating_value = review_rating.get("ratingValue", "N/A")
+            alternate_name = review_rating.get("alternateName", "N/A")
+            review_body = inner_json.get("reviewBody", "N/A")
 
-else:
-    st.error("Error in fact-checking: " + api_response['error'])
+            # Display the results
+            if rating_value.isdigit():
+                st.progress(int(rating_value) / 5, "Fact-checking in progress.")
+            else:
+                st.write("Rating Value: ", rating_value)
+            st.write(f"**Review Body**: {review_body}")
+        
+            # Create a two-column layout
+            col1, col2 = st.columns(2)
+        
+            # Use the metric widget to display the rating and alternate name
+            col1.metric("Rating", rating_value)
+            col2.metric("Verdict", alternate_name)
+            
+            #with st.expander("Here is the final JSON-LD"):
+            #    st.json(api_response)  # Display the entire JSON-LD data
+                except json.JSONDecodeError as e:
+            st.error(f"Error decoding JSON: {e}")
+    else:
+        st.error("Error in fact-checking: " + api_response['error'])