Spaces:
Running
Running
Update Summarization/main.py
Browse files- 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)
|