Update app/main.py

app/main.py

@@ -1,6 +1,8 @@
-# main.py
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import JSONResponse
+# app/main.py
+from fastapi import FastAPI, File, UploadFile, HTTPException, Request, Form
+from fastapi.responses import JSONResponse, StreamingResponse, HTMLResponse, RedirectResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from typing import Dict, Any
 import io
 import base64
@@ -17,20 +19,24 @@ app = FastAPI(
     version="1.0.0"
 )
 
-@app.get("/", summary="Root Endpoint")
-def read_root():
+# Mount static files if any (optional)
+# app.mount("/static", StaticFiles(directory="app/static"), name="static")
+
+# Set up templates
+templates = Jinja2Templates(directory="app/templates")
+
+@app.get("/", response_class=HTMLResponse, summary="Root Endpoint")
+def read_root(request: Request):
     """
-    Root endpoint to verify that the API is running.
+    Render the main page with upload form.
     """
     logger.info("Root endpoint accessed.")
-    return {"message": "PDF Language Issue Analyzer API is running."}
+    return templates.TemplateResponse("index.html", {"request": request})
 
-@app.post("/analyze", summary="Analyze PDF for Language Issues")
-async def analyze_pdf_endpoint(file: UploadFile = File(...)):
+@app.post("/analyze", response_class=HTMLResponse, summary="Analyze PDF for Language Issues")
+async def analyze_pdf_endpoint(request: Request, file: UploadFile = File(...)):
     """
-    Analyze an uploaded PDF file for language issues.
-
-    - **file**: PDF file to be analyzed.
+    Analyze an uploaded PDF file for language issues and render the results on the frontend.
     """
     if file.content_type != "application/pdf":
         logger.error("Invalid file type uploaded: %s", file.content_type)
@@ -41,18 +47,20 @@ async def analyze_pdf_endpoint(file: UploadFile = File(...)):
         file_stream = io.BytesIO(contents)
         language_issues, annotated_pdf = analyze_pdf(file_stream)
 
-        response = {
+        response_context = {
+            "request": request,
             "language_issues": language_issues
         }
 
         if annotated_pdf:
-            # Properly encode the annotated PDF in Base64
+            # Encode the annotated PDF in Base64
             encoded_pdf = base64.b64encode(annotated_pdf).decode('utf-8')
-            response["annotated_pdf"] = f"data:application/pdf;base64,{encoded_pdf}"
+            annotated_pdf_data_url = f"data:application/pdf;base64,{encoded_pdf}"
+            response_context["annotated_pdf"] = annotated_pdf_data_url
             logger.info("Annotated PDF generated and encoded.")
 
         logger.info("PDF analysis completed successfully.")
-        return JSONResponse(content=response)
+        return templates.TemplateResponse("index.html", response_context)
     except Exception as e:
         logger.exception("Error occurred during PDF analysis.")
         raise HTTPException(status_code=500, detail=str(e))