Update app.py

app.py

@@ -1,37 +1,18 @@
-from fastapi import FastAPI, UploadFile, File
-from fastapi.responses import JSONResponse
-from fastapi.middleware.cors import CORSMiddleware
-import tempfile, shutil, os
-import pandas as pd
+import gradio as gr
+import tempfile, os
 from pipeline import run_pipeline
-
-app = FastAPI(title="SC API", version="1.0")
-
-# Allow CORS so any frontend can call the API
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+import pandas as pd
 
 DATASET_PATH = "dataset.csv"
 
-@app.post("/analyze-pdfs/")
-async def analyze_pdfs(files: list[UploadFile] = File(...)):
-    """
-    Upload PDFs → process with SC pipeline → return JSON → save to dataset.csv
-    """
+def analyze_pdf(file):
     with tempfile.TemporaryDirectory() as tmpdirname:
-        for file in files:
-            file_path = os.path.join(tmpdirname, file.filename)
-            with open(file_path, "wb") as buffer:
-                shutil.copyfileobj(file.file, buffer)
-
+        file_path = os.path.join(tmpdirname, file.name)
+        with open(file_path, "wb") as f:
+            f.write(file.read())
         results = run_pipeline(tmpdirname)
-        json_result = results.to_dict(orient="records")
 
-        # Save results to CSV
+        # Save to dataset.csv
         if os.path.exists(DATASET_PATH):
             dataset = pd.read_csv(DATASET_PATH)
             dataset = pd.concat([dataset, results], ignore_index=True)
@@ -39,4 +20,13 @@ async def analyze_pdfs(files: list[UploadFile] = File(...)):
             dataset = results
         dataset.to_csv(DATASET_PATH, index=False)
 
-        return JSONResponse(content={"results": json_result})
+    return results.to_dict(orient="records")
+
+iface = gr.Interface(
+    fn=analyze_pdf,
+    inputs=gr.File(file_types=[".pdf"]),
+    outputs=gr.JSON(),
+    title="SC API"
+)
+
+iface.launch()