Update app.py

app.py

@@ -2,17 +2,17 @@ from fastapi import FastAPI, UploadFile, File
 from pydantic import BaseModel
 from groq import Groq
 from pypdf import PdfReader
+import io
+import uvicorn
 
 app = FastAPI()
 
 # Initialize Groq client
-client = Groq(api_key="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")
+client = Groq(api_key="YOUR_API_KEY_HERE")
 
-# -----------------------------
 # Extract text from PDF
-# -----------------------------
-def extract_pdf_text(pdf_file):
-    reader = PdfReader(pdf_file)
+def extract_pdf_text(pdf_bytes):
+    reader = PdfReader(io.BytesIO(pdf_bytes))
     text = ""
     for page in reader.pages:
         extracted = page.extract_text()
@@ -20,9 +20,7 @@ def extract_pdf_text(pdf_file):
             text += extracted + "\n"
     return text
 
-# -----------------------------
 # Analyze text with Groq
-# -----------------------------
 def analyze_text_with_groq(text):
     prompt = f"""
 You are an expert document analysis AI.
@@ -50,16 +48,13 @@ Return your answer in this JSON structure:
 
     return response.choices[0].message.content
 
-# -----------------------------
-# FastAPI Endpoint
-# -----------------------------
 @app.post("/analyze")
 async def analyze_document(file: UploadFile = File(...)):
-    # Extract text from uploaded PDF
     pdf_bytes = await file.read()
     text = extract_pdf_text(pdf_bytes)
-
-    # Analyze with Groq
     result = analyze_text_with_groq(text)
+    return {"result": result}
 
-    return {"result": result}
+# REQUIRED for Hugging Face Docker Spaces
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)