Add fastapi endpoint

main.py

@@ -1,8 +1,10 @@
 import os
 import openai
+from fastapi import FastAPI
 
-openai.api_key = os.environ.get("openai_api_key")
 
+openai.api_key = os.environ.get("openai_api_key")
+app = FastAPI()
 
 _tweet = "I'm a pretty average middle aged guy, got a solid job and had a good family but things got messy and took " \
          "a turn for the worst. Everyone says you need to hit rock bottom to go back up so here I am!"
@@ -12,15 +14,15 @@ def prompt(tweet, roast=True):
     roast_or_toast = "snarky joke" if roast else "encouragement"
     return f"A user tweeted this tweet:\n\n{tweet}\\n\nThe following {roast_or_toast} was given as an answer:"
 
-
-response = openai.Completion.create(
-    engine="text-davinci-002",
-    prompt=prompt(_tweet, roast=True),
-    temperature=0.7,
-    max_tokens=60,
-    top_p=1,
-    frequency_penalty=0,
-    presence_penalty=0
-)
-
-print(response.choices[0].text.strip())
+@app.get("/")
+def read_root():
+    response = openai.Completion.create(
+        engine="text-davinci-002",
+        prompt=prompt(_tweet, roast=True),
+        temperature=0.7,
+        max_tokens=60,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+    )
+    return response.choices[0].text.strip()