@app.post("/api/ask", response_model=ResponseModel)
def get_response(user_input: UserInput):
    try:
    
        # Load .env file for each request and overwrite existing variables
        load_dotenv(override=True)

        # Get BARD_API_KEY from the .env file
        BARD_API_KEY = os.environ.get('BARD_API_KEY')

        # Set up the session for each request
        session = requests.Session()
        session.headers = {
            "Host": "gemini.google.com",
            "X-Same-Domain": "1",
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36",
            "Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
            "Origin": "https://gemini.google.com",
            "Referer": "https://gemini.google.com/",
        }
        session.cookies.set("__Secure-1PSID", BARD_API_KEY)

        bard = Bard(token=BARD_API_KEY, session=session, timeout=30)
            
        # Print the user input to the console
        print(f"Received user input: {user_input.user_input}")    
        response = process_input(user_input.user_input, bard)
        # Save response to a temporary file
        temp_file_path = save_response_to_file(response)
        # Return the path of the temporary file instead of the response content
        return {"content": temp_file_path}        
        
        print(f"Response saved to: {temp_file_path}")  # For debugging
        return response
    except Exception as e:
        # Log the error details for debugging
        print(f"Error: {e}")
        # You might want to return a more generic error message in a production environment
        raise HTTPException(status_code=400, detail=f"Error processing request: {e}")
        
def save_response_to_file(response):
    # Define the desired directory path
    directory_path = '/www/html/temp/'

    # Ensure the directory exists, if not, create it
    os.makedirs(directory_path, exist_ok=True)

    # Create a unique file name. You can also use other methods to generate unique names
    file_name = f"response_{int(time.time())}.json"
    file_path = os.path.join(directory_path, file_name)

    # Write the response to the file
    with open(file_path, 'w') as file:
        json.dump(response, file)

    return file_path      

def process_input(user_input, bard):
    response = bard.get_answer(user_input)
    result = {
        'content': response['content'],
        'images': list(response['images']) if 'images' in response and isinstance(response['images'], set) else response.get('images', []),
        'links': list(response['links']) if 'links' in response and isinstance(response['links'], set) else response.get('links', [])
    }

    return result

pip install -q -U python-gemini-api