ikraamkb commited on
Commit
064b85f
·
verified ·
1 Parent(s): ea75150

Update Summarization/main.py

Browse files
Files changed (1) hide show
  1. Summarization/main.py +0 -9
Summarization/main.py CHANGED
@@ -1,6 +1,5 @@
1
  from fastapi import FastAPI, UploadFile, File, Form, Request
2
  from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
3
- from fastapi.staticfiles import StaticFiles
4
  from fastapi.templating import Jinja2Templates
5
  from fastapi.middleware.cors import CORSMiddleware
6
  import tempfile, os
@@ -18,36 +17,28 @@ app.add_middleware(
18
  allow_headers=["*"],
19
  )
20
 
21
- app.mount("/static", StaticFiles(directory="static"), name="static")
22
- app.mount("/resources", StaticFiles(directory="resources"), name="resources")
23
  templates = Jinja2Templates(directory="templates")
24
 
25
- # Serve homepage
26
  @app.get("/", response_class=HTMLResponse)
27
  async def serve_home(request: Request):
28
  return templates.TemplateResponse("HomeS.html", {"request": request})
29
 
30
- # Summarize endpoint (calls summarize_document from app.py)
31
  @app.post("/summarize/")
32
  async def summarize_api(file: UploadFile = File(...), length: str = Form("medium")):
33
  try:
34
  result = await summarize_document(file, length)
35
  return JSONResponse(result)
36
  except Exception as e:
37
- print(f"Error during summarization: {str(e)}")
38
  return JSONResponse({"detail": f"Internal server error: {str(e)}"}, status_code=500)
39
 
40
- # Captioning endpoint
41
  @app.post("/imagecaption/")
42
  async def caption(file: UploadFile = File(...)):
43
  try:
44
  result = await caption_image(file)
45
  return JSONResponse(result)
46
  except Exception as e:
47
- print(f"Error during captioning: {str(e)}")
48
  return JSONResponse({"error": f"Image captioning failed: {str(e)}"}, status_code=500)
49
 
50
- # Serve audio/pdf files
51
  @app.get("/files/{filename}")
52
  async def serve_file(filename: str):
53
  filepath = os.path.join(tempfile.gettempdir(), filename)
 
1
  from fastapi import FastAPI, UploadFile, File, Form, Request
2
  from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
 
3
  from fastapi.templating import Jinja2Templates
4
  from fastapi.middleware.cors import CORSMiddleware
5
  import tempfile, os
 
17
  allow_headers=["*"],
18
  )
19
 
 
 
20
  templates = Jinja2Templates(directory="templates")
21
 
 
22
  @app.get("/", response_class=HTMLResponse)
23
  async def serve_home(request: Request):
24
  return templates.TemplateResponse("HomeS.html", {"request": request})
25
 
 
26
  @app.post("/summarize/")
27
  async def summarize_api(file: UploadFile = File(...), length: str = Form("medium")):
28
  try:
29
  result = await summarize_document(file, length)
30
  return JSONResponse(result)
31
  except Exception as e:
 
32
  return JSONResponse({"detail": f"Internal server error: {str(e)}"}, status_code=500)
33
 
 
34
  @app.post("/imagecaption/")
35
  async def caption(file: UploadFile = File(...)):
36
  try:
37
  result = await caption_image(file)
38
  return JSONResponse(result)
39
  except Exception as e:
 
40
  return JSONResponse({"error": f"Image captioning failed: {str(e)}"}, status_code=500)
41
 
 
42
  @app.get("/files/{filename}")
43
  async def serve_file(filename: str):
44
  filepath = os.path.join(tempfile.gettempdir(), filename)