import streamlit as st
import requests

# Create a session for reusing connections
session = requests.Session()

# Function to interact with the AI
def chat_with_ai(message):
    api_url = "https://free-ai-api.devastation-war.repl.co/chat"
    payload = {"message": message}

    try:
        with session.post(api_url, json=payload) as response:
            if response.status_code == 200:
                return response.json().get('response')  # Access 'response' key
            else:
                return {"error": "Failed to get a response from the AI API. 😞"}
    except requests.RequestException as e:
        return {"error": f"Error: {e}"}

# Function to perform web search
def web_search(query):
    url = "https://free-ai-api.devastation-war.repl.co/search"
    payload = {"query": query}
    response = requests.post(url, json=payload)

    if response.status_code == 200:
        return response.json()
    else:
        return {"error": f"Error: {response.status_code}"}

# Main function
def main():
    st.title("🔍 HelpingAI Research Assistant")
    st.sidebar.header("🛠️ Settings")
    query = st.sidebar.text_input("🔎 Enter your research query: ")
    generate_report = st.sidebar.button("📝 Generate Report")

    if generate_report:
        if query:
            with st.spinner('🔄 Searching...'):
                # Perform web search
                search_results = web_search(query)
            
            # Pass the search results to the AI for generating a report
            prompt = f"Generate a research report based on the following information: {search_results}. If the search results are insufficient, answer the user's question using the information available."
            with st.spinner('🔄 Generating report...'):
                report = chat_with_ai(prompt)
            
            # Display the report
            st.write(report)
        else:
            st.sidebar.error("❗ Please enter a research query.")

if __name__ == "__main__":
    main()