Abhaykoul's picture
Update app.py
47e5b50
raw
history blame
No virus
2.13 kB
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().get('results') # Assuming 'results' contain relevant info
else:
return {"error": f"Error: {response.status_code}"}
# Main function
def main():
query = st.text_area("Enter resume content:")
if query:
prompt = f"Please provide a comprehensive review of the following resume and suggest improvements. The goal is to enhance its effectiveness in conveying the candidate's skills and experiences. Here's the resume content: {query}"
# Perform web search
search_results = web_search(query)
# Extract relevant information and convert it into a message
message = "" # Create an empty string to store the message
for result in search_results:
# Assuming each result is a string, you might adjust this part accordingly
message += result + ". " # Concatenate results with a period
# Combine prompt and message for AI conversation
full_message = prompt + " " + message
# Pass the combined message to the AI for generating a report
report = chat_with_ai(full_message)
# Display the report
st.write(report)
if __name__ == "__main__":
main()