import streamlit as st
import requests
import json
import os

# Define fact_check_statement function here...
def fact_check_statement(query, api_key):
    url = "https://api.wordlift.io/fact-check/score"

    payload = json.dumps({"query": query})
    headers = {
        "Content-Type": "application/json",
        "Accept": "application/json",
        "Authorization": "Key " + api_key,
    }

    try:
        response = requests.request("POST", url, headers=headers, data=payload)
        logger.info(
            f"Request made to API: {url} with payload: {payload} and headers: {headers}"
        )
        if response.status_code == 200:
            return response.json()
        else:
            logger.error(
                f"API request failed with status code {response.status_code} and response: {response.text}"
            )
            return {
                "error": f"Failed to get response from the fact-checking API. Status code: {response.status_code}"
            }
    except Exception as e:
        logger.error(f"Exception during API request: {e}")
        return {"error": f"Exception during API request: {str(e)}"}

# Page config
st.set_page_config(
    page_title="AI Fact-Checking by WordLift",
    page_icon="fav-ico.png",
    layout="wide",
    initial_sidebar_state="collapsed",
    menu_items={
        'Get Help': 'https://wordlift.io/book-a-demo/',
        'About': "# This is a demo app for AI Fact-Checking"
    }
)

# Sidebar
st.sidebar.image("logo-wordlift.png")

# Main content
with st.form(key='my_form'):
    text_input = st.text_area(label='Enter a statement for fact-checking')
    submit_button = st.form_submit_button(label='Check Facts')

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:
        # Parse the outer JSON
        inner_json_str = api_response.get("response", "{}")
        # Parse the inner JSON string
        inner_json = json.loads(inner_json_str)

        # Extract relevant information
        claim_reviewed = inner_json.get("claimReviewed", "N/A")
        review_rating = inner_json.get("reviewRating", {})
        rating_value = review_rating.get("ratingValue", "N/A")
        review_body = inner_json.get("reviewBody", "N/A")

        # Display the results
        st.write(f"Claim Reviewed: {claim_reviewed}")
        st.write(f"Review Body: {review_body}")
        st.json(inner_json)  # Display the entire JSON-LD data

        # Assuming the rating value is a number between 1 and 5
        if rating_value.isdigit():
            st.progress(int(rating_value) / 5)
        else:
            st.write("Rating Value: ", rating_value)
    else:
        st.error("Error in fact-checking: " + api_response['error'])