Update app.py

app.py

@@ -4,8 +4,21 @@ import os
 import tempfile
 from pathlib import Path
 
-# Import your pipeline function
+# NOTE: You must ensure that 'working_yolo_pipeline.py' exists 
+# and defines the following items correctly:
 from working_yolo_pipeline import run_document_pipeline, DEFAULT_LAYOUTLMV3_MODEL_PATH, WEIGHTS_PATH
+# Since I don't have this file, I am assuming the imports are correct.
+
+# Define placeholders for assumed constants if the pipeline file isn't present
+# You should replace these with your actual definitions if they are missing
+try:
+    from working_yolo_pipeline import run_document_pipeline, DEFAULT_LAYOUTLMV3_MODEL_PATH, WEIGHTS_PATH
+except ImportError:
+    print("Warning: 'working_yolo_pipeline.py' not found. Using dummy paths.")
+    def run_document_pipeline(*args):
+        return {"error": "Placeholder pipeline function called."}
+    DEFAULT_LAYOUTLMV3_MODEL_PATH = "./models/layoutlmv3_model"
+    WEIGHTS_PATH = "./weights/yolo_weights.pt"
 
 
 def process_pdf(pdf_file, layoutlmv3_model_path=None):
@@ -47,16 +60,13 @@ def process_pdf(pdf_file, layoutlmv3_model_path=None):
         output_filename = f"{Path(pdf_path).stem}_analysis.json"
         temp_output = tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.json', prefix='analysis_')
 
+        # Dump results to the temporary file
         with open(temp_output.name, 'w', encoding='utf-8') as f:
             json.dump(result, f, indent=2, ensure_ascii=False)
 
         # Format JSON for display
         json_display = json.dumps(result, indent=2, ensure_ascii=False)
 
-        # Truncate if too long for display
-        # if len(json_display) > 50000:
-        #     json_display = json_display[:50000] + "\n\n... (truncated for display, download full file)"
-
         return json_display, temp_output.name
 
     except Exception as e:
@@ -64,7 +74,8 @@ def process_pdf(pdf_file, layoutlmv3_model_path=None):
 
 
 # Create Gradio interface
-with gr.Blocks(title="Document Analysis Pipeline", theme=gr.themes.Soft()) as demo:
+# FIX APPLIED: Removed 'theme=gr.themes.Soft()' which caused the TypeError
+with gr.Blocks(title="Document Analysis Pipeline") as demo:
     gr.Markdown("""
     # 📄 Document Analysis Pipeline