Aditya12kumar commited on
Commit
893f7a9
·
verified ·
1 Parent(s): f659e62

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +16 -0
main.py CHANGED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from .QuoteGenerator import QuoteGenerator
3
+
4
+ app = FastAPI()
5
+
6
+ quote_generator = QuoteGenerator()
7
+ quote_generator.load_generator()
8
+
9
+ @app.get("/")
10
+ def root():
11
+ return {"message": "This is Quote Generation API test"}
12
+
13
+ @app.post("/generate_quote")
14
+ def generate_quote_for_user(prompt):
15
+ generated_quote = quote_generator.generate_quote(prompt)
16
+ return generated_quote