Update app.py

app.py

@@ -1,24 +1,54 @@
-from fastapi import FastAPI
+# from fastapi import FastAPI
+# from transformers import pipeline
+
+# ## create a new FASTAPI app instance
+# app=FastAPI()
+
+# # Initialize the text generation pipeline
+# pipe = pipeline("text2text-generation", model="google/flan-t5-small")
+
+
+# @app.get("/")
+# def home():
+#     return {"message":"Hello World"}
+
+# # Define a function to handle the GET request at `/generate`
+
+
+# @app.get("/generate")
+# def generate(text:str):
+#     ## use the pipeline to generate text from given input text
+#     output=pipe(text)
+
+#     ## return the generate text in Json reposne
+#     return {"output":output[0]['generated_text']}
+
+from fastapi import FastAPI, Request, Form
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from transformers import pipeline
 
-## create a new FASTAPI app instance
-app=FastAPI()
+# Create a new FastAPI app instance
+app = FastAPI()
 
 # Initialize the text generation pipeline
 pipe = pipeline("text2text-generation", model="google/flan-t5-small")
 
+# Mount the static files directory
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
-@app.get("/")
-def home():
-    return {"message":"Hello World"}
-
-# Define a function to handle the GET request at `/generate`
+# Initialize the templates directory
+templates = Jinja2Templates(directory="templates")
 
+@app.get("/", response_class=HTMLResponse)
+async def home(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
 
-@app.get("/generate")
-def generate(text:str):
-    ## use the pipeline to generate text from given input text
-    output=pipe(text)
+@app.post("/generate", response_class=HTMLResponse)
+async def generate(request: Request, text: str = Form(...)):
+    # Use the pipeline to generate text from given input text
+    output = pipe(text)
 
-    ## return the generate text in Json reposne
-    return {"output":output[0]['generated_text']}
+    # Return the generated text in the template response
+    return templates.TemplateResponse("index.html", {"request": request, "output": output[0]['generated_text']})