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://test.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.write("Hello! I am HelpingAI, developed by Abhay Koul. My purpose is to simplify your learning journey by providing personalized assistance, innovative teaching methods, and tailored resources to meet your unique needs. I am here to make your educational experience more enjoyable and effective. Feel free to ask me any questions or let me know how I can assist you in your learning adventure and in many more things from your life.")
    
    query = st.text_input("Enter your research query: ")
    
    if st.button("Generate Report"):
        if query:
            # 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 text: {search_results}"
            report = chat_with_ai(prompt)
            
            # Display the report
            st.write(report)
        else:
            st.write("Please enter a research query.")

if __name__ == "__main__":
    main()