Initial Commit

source/app.py

@@ -0,0 +1,33 @@
+import calendar_connection as cal
+import os
+from flask import Flask, request, render_template
+import json
+import requests
+
+app = Flask(__name__)
+
+@app.route('/')
+def index():
+    return render_template("index.html")
+
+@app.route('/calendar/getCalendarId', methods = ['POST'])
+def get_calendar():
+    return json.dumps({"calendar_id": cal.get_calendar(request.json["token"], request.json["timezone"])})
+
+def get_json(input):
+    headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
+    with open("prompt.txt", "r") as promptFile:
+        payload = {
+            "inputs": f"Here is a set of sample prompts, which are only to be used for reference:\n{promptFile.read()}\nWhat is the output to {input}? Only return one line of output.\nOutput: ",
+            "parameters": {"temperature": 1, "return_full_text": False, "top_k": 1}
+        }
+    url = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+    response = requests.post(url = url, headers=headers, json=payload).json()[0]["generated_text"]
+    return response
+
+@app.route('/calendar/execute', methods = ['POST'])
+def calendar():
+    return json.dumps({"output": cal.execute_json(request.json["token"], request.json["calendar_id"], request.json["timezone"], get_json(request.json["input"]))})
+
+if __name__ == "__main__":
+    app.run()