Update main.py

main.py

@@ -4,8 +4,18 @@ openai.api_key = "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"
 
 app = FastAPI()
 
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+
 @app.get("/")
-def read_root():
-    input = {"role": "user", "content": "Translate this sentence to French. Sentence: I can speak Chinese."}
-    output = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[input])
-    return {"Input": input, "Output": output}
+def index() -> FileResponse:
+    return FileResponse(path="/app/static/index.html", media_type="text/html")
+
+@app.get("/infer_chatgpt")
+def chatgpt(input):
+    try:
+        prompt = {"role": "user", "content": input}
+        result = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[prompt])
+        output = result.get('choices')[0].get('message').get('content')
+    except Exception as e:
+        output = str(e)
+    return {"output": output}