Update app.py

app.py

@@ -1,41 +1,18 @@
-# -*- coding: utf-8 -*-
-"""API prog
-
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1dwnXx2hUYegX54dlWrXEF8uF_Jj7QxX5
-"""
-
-from fastapi import FastAPI
+from fastapi import FastAPI, UploadFile, File
 from pydantic import BaseModel
 from groq import Groq
+from pypdf import PdfReader
 
 app = FastAPI()
 
-class DocInput(BaseModel):
-    text: str
-
-client = Groq(api_key="YOUR_KEY")
-
-@app.post("/analyze")
-def analyze_document(data: DocInput):
-    return {"result": "ok"}
-from groq import Groq
-from pypdf import PdfReader
-import gradio as gr
-
-# ---------------------------------------
-# 1. Initialize Groq Client
-# ---------------------------------------
+# Initialize Groq client
 client = Groq(api_key="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")
 
-
-# ---------------------------------------
-# 2. Extract text from PDF
-# ---------------------------------------
+# -----------------------------
+# Extract text from PDF
+# -----------------------------
 def extract_pdf_text(pdf_file):
-    reader = PdfReader(pdf_file.name)
+    reader = PdfReader(pdf_file)
     text = ""
     for page in reader.pages:
         extracted = page.extract_text()
@@ -43,16 +20,13 @@ def extract_pdf_text(pdf_file):
             text += extracted + "\n"
     return text
 
-
-# ---------------------------------------
-# 3. Ask Groq to classify + extract fields
-# ---------------------------------------
-def analyze_document(text):
+# -----------------------------
+# Analyze text with Groq
+# -----------------------------
+def analyze_text_with_groq(text):
     prompt = f"""
 You are an expert document analysis AI.
-
 Given the following document text, do 3 things:
-
 1. Identify the document type (invoice, receipt, contract, report, certificate, etc.)
 2. Extract key fields in JSON format.
 3. Provide a short summary.
@@ -61,7 +35,6 @@ Document text:
 {text}
 
 Return your answer in this JSON structure:
-
 {{
   "document_type": "",
   "fields": {{}},
@@ -75,4 +48,18 @@ Return your answer in this JSON structure:
         temperature=0
     )
 
-    return response.choices[0].message.content
+    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}