Create app.py

app.py

@@ -0,0 +1,42 @@
+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'))
+