Update app.py

app.py

@@ -409,32 +409,43 @@ def identify_headers_with_openrouter(pdf_path, model, LLM_prompt, pages_to_check
 
 
 def identify_headers_and_save_excel(pdf_path, model, llm_prompt):
-    # Call your existing function
-    result = identify_headers_with_openrouter(pdf_path, model, llm_prompt)
-    
-    if not result:
-        return None
-    df = pd.DataFrame(result)
-    
-    # Save Excel to a file on disk
-    output_path = "output.xlsx"
-    df.to_excel(output_path, index=False, engine='openpyxl')
+    try:
+        # 1. Get the result from your LLM function
+        result = identify_headers_with_openrouter(pdf_path, model, llm_prompt)
+        
+        # 2. Safety Check: If LLM failed or returned nothing
+        if not result:
+            logger.warning("No headers found or LLM failed. Creating an empty report.")
+            df = pd.DataFrame([{"System Message": "No headers were identified by the LLM."}])
+        else:
+            df = pd.DataFrame(result)
+        
+        # 3. Use an Absolute Path for the output
+        # This ensures Gradio knows exactly where the file is
+        output_path = os.path.abspath("header_analysis_output.xlsx")
+        
+        # 4. Save using the engine explicitly
+        df.to_excel(output_path, index=False, engine='openpyxl')
+        
+        logger.info(f"File successfully saved to {output_path}")
+        return output_path 
 
-    
-    return output_path  # return file path, not BytesIO
+    except Exception as e:
+        logger.error(f"Critical error in processing: {str(e)}")
+        # Return None or a custom error message to Gradio
+        return None
 
+# Improved launch with debug mode enabled
 iface = gr.Interface(
     fn=identify_headers_and_save_excel,
     inputs=[
-        gr.Textbox(label="PDF URL"),
-        gr.Textbox(label="Model Type"),
-        gr.Textbox(label="LLM Prompt")
+        gr.Textbox(label="PDF URL", placeholder="https://example.com/file.pdf"),
+        gr.Textbox(label="Model Type", value="google/gemini-2.0-flash-001"), # Default example
+        gr.Textbox(label="LLM Prompt", lines=3)
     ],
-    outputs=gr.File(label="Download Excel")
+    outputs=gr.File(label="Download Excel Results"),
+    title="PDF Header Extractor"
 )
-if __name__ == "__main__":
-    try:
-        iface.close() # Close existing instance if it exists
-    except:
-        pass
-    iface.launch()
+
+# Launch with debug=True to see errors in the console
+iface.launch(debug=True)