Update app.py

app.py

@@ -212,14 +212,31 @@ def identify_headers_with_openrouter(pdf_path, model,LLM_prompt, pages_to_check=
 
     return out
 
+# Wrapper function to convert JSON to a dataframe-friendly format
+def identify_headers_with_table(pdf_path, model, LLM_prompt):
+    # Call your existing function
+    result = identify_headers_with_openrouter(pdf_path, model, LLM_prompt)
+    
+    # Convert list of dicts to list of lists for Gradio Dataframe
+    if not result:
+        return []  # empty table if no results
+    
+    table_data = [[item['text'], item['page']+1, item['suggested_level'], item['confidence']] for item in result]
+    return table_data
+
+# Column names for the table
+columns = ["Text", "Page", "Suggested Level", "Confidence"]
 
 
+# Gradio Interface
 iface = gr.Interface(
-    fn=identify_headers_with_openrouter,
-    inputs=[gr.Textbox(label="Document Link"),
+    fn=identify_headers_with_table,
+    inputs=[
+        gr.Textbox(label="Document Link"),
         gr.Textbox(label="Model Type"),
-        gr.Textbox(label="LLM Prompt")]
-    outputs=gr.Textbox(label="Output")
+        gr.Textbox(label="LLM Prompt")
+    ],
+    outputs=gr.Dataframe(headers=columns)
 )
 
 iface.launch()