Update app.py

app.py

@@ -1,27 +1,48 @@
 import fitz  # PyMuPDF
-import os
 import json
 import gradio as gr
 from groq import Groq
+from dotenv import load_dotenv
+import os
+
+# Load environment variables
+load_dotenv()
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# 🔐 Initialize Groq Client
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# Initialize Groq Client
+client = Groq(api_key=GROQ_API_KEY)
 
 def extract_form_fields(pdf_bytes):
-    doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+    if not pdf_bytes or len(pdf_bytes) == 0:
+        raise ValueError("Uploaded file is empty or not a valid PDF.")
+
+    try:
+        doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+    except Exception as e:
+        raise ValueError(f"Failed to open PDF: {e}")
+
     form_fields = {}
     for page in doc:
-        widgets = page.widgets()
-        if widgets:
-            for widget in widgets:
-                key = widget.field_name
-                value = widget.field_value if widget.field_value else ""
-                form_fields[key] = value
+        for widget in page.widgets():
+            key = widget.field_name
+            value = widget.field_value if widget.field_value else ""
+            form_fields[key] = value
+
     return form_fields
 
 def get_pdf_text(pdf_bytes):
-    doc = fitz.open(stream=pdf_bytes, filetype="pdf")
-    return "\n".join([page.get_text() for page in doc])
+    if not pdf_bytes or len(pdf_bytes) == 0:
+        raise ValueError("Uploaded file is empty or not a valid PDF.")
+
+    try:
+        doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+    except Exception as e:
+        raise ValueError(f"Failed to open PDF: {e}")
+
+    text = ""
+    for page in doc:
+        text += page.get_text()
+    return text
 
 def get_field_details(form_fields, pdf_text):
     prompt = f"""
@@ -36,19 +57,26 @@ Based on this, explain the meaning or expected value of each of the following fi
 Return your output in the following JSON format:
 {{ "field_name_1": "description", "field_name_2": "description", ... }}
     """
+
     response = client.chat.completions.create(
         model="llama3-8b-8192",
         messages=[{"role": "user", "content": prompt}]
     )
 
-    return response.choices[0].message.content
+    try:
+        explanation = response.choices[0].message.content
+        return explanation
+    except Exception as e:
+        return f"Failed to analyze fields: {str(e)}"
 
 def analyze_form(pdf_file):
     if not pdf_file:
         return "❌ No file provided.", "", ""
 
     try:
-        pdf_bytes = pdf_file.read()
+        with open(pdf_file.name, 'rb') as f:
+            pdf_bytes = f.read()
+
         fields = extract_form_fields(pdf_bytes)
         pdf_text = get_pdf_text(pdf_bytes)
         explanation = get_field_details(fields, pdf_text)
@@ -57,19 +85,21 @@ def analyze_form(pdf_file):
     except Exception as e:
         return f"❌ Error: {str(e)}", "", ""
 
-# 🎨 Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("## 📄 Form Field Analyzer")
-    gr.Markdown("Upload a **tax or registration PDF form**. This tool extracts fillable fields and describes what each might represent.")
-
-    with gr.Row():
-        file_input = gr.File(label="Upload PDF", file_types=[".pdf"])
-        analyze_btn = gr.Button("Analyze Form")
-
-    status_output = gr.Textbox(label="Status")
-    fields_output = gr.Code(label="📋 Extracted Fields (JSON)", language="json")
-    desc_output = gr.Code(label="💡 Field Descriptions (LLM Response)", language="json")
-
-    analyze_btn.click(fn=analyze_form, inputs=[file_input], outputs=[status_output, fields_output, desc_output])
-
-demo.launch()
+# Gradio Interface
+description = "📄 Upload a tax or registration form PDF. This tool extracts form fields and explains what each one likely means or requires using LLM."
+
+iface = gr.Interface(
+    fn=analyze_form,
+    inputs=gr.File(label="Upload PDF", type="filepath"),
+    outputs=[
+        gr.Textbox(label="Status"),
+        gr.Code(label="📋 Extracted Form Fields (JSON)", language="json"),
+        gr.Code(label="💡 Field Descriptions (JSON)", language="json"),
+    ],
+    title="📄 Form Field Analyzer",
+    description=description,
+    theme="default"
+)
+
+if __name__ == "__main__":
+    iface.launch()