import asyncio
import json
from flask_cors import CORS
from flask import Flask, request, jsonify
from waitress import serve
import concurrent.futures
import requests

import os
import google.generativeai as genai

app = Flask(__name__)
app.config['JSON_AS_ASCII'] = False
CORS(app, resources={r"/*": {"origins": "*"}})

API_KEY = os.environ['API_KEY']
genai.configure(api_key=API_KEY)
model = genai.GenerativeModel('gemini-pro')

def get_response(prompt:str):
    try:
        response = model.generate_content(prompt)
        return response
    except Exception as e:
        return str(e)

def create_handler(name: str):
    def handler():
        try:
            if name == 'write':
                response = get_response(prompt)
                return response

        except Exception as error:
            return jsonify({"error": str(error)}), 500

    handler.__name__ = name
    return handler


app.route('/write', methods=['POST'])(create_handler('write'))