Update app.py
Browse files
app.py
CHANGED
@@ -1,114 +1,119 @@
|
|
1 |
# app.py
|
2 |
import json
|
3 |
import traceback
|
4 |
-
import tempfile
|
5 |
import os
|
6 |
import gradio as gr
|
7 |
-
from typing import Tuple, Optional, Any
|
8 |
|
9 |
# Import config first to ensure JAVA_HOME is set early
|
10 |
import config
|
11 |
|
12 |
from main_analyzer import analyze_pdf
|
13 |
-
# language_tool_python
|
14 |
-
import language_tool_python
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
|
17 |
-
def process_upload(uploaded_file_input: Optional[Any]) -> Tuple[str, Optional[str]]:
|
18 |
-
"""
|
19 |
-
Processes the uploaded file from Gradio.
|
20 |
-
'uploaded_file_input' is expected to be an object from gr.File(),
|
21 |
-
which could be a path string (like gradio.utils.NamedString) or a file-like object.
|
22 |
-
"""
|
23 |
-
if uploaded_file_input is None:
|
24 |
-
print("App: No file uploaded.")
|
25 |
-
return json.dumps({"error": "No file uploaded."}, indent=2), None
|
26 |
|
27 |
try:
|
28 |
-
|
29 |
-
#
|
30 |
-
#
|
31 |
-
print(f"App: Received file input of type: {type(uploaded_file_input)}. Passing to analyzer.")
|
32 |
|
33 |
-
|
|
|
|
|
34 |
|
35 |
-
#
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
|
|
|
|
|
|
40 |
|
41 |
results_json = json.dumps(results_dict, indent=2, ensure_ascii=False)
|
42 |
return results_json, None
|
43 |
|
44 |
except Exception as e:
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
return error_message, None
|
54 |
-
# No
|
55 |
-
# Gradio
|
56 |
-
|
57 |
|
58 |
def create_interface():
|
59 |
with gr.Blocks(title="PDF Analyzer") as interface:
|
60 |
-
gr.Markdown("# PDF Analyzer")
|
61 |
-
gr.Markdown(
|
62 |
-
"Upload a PDF document to analyze its structure, references, language, and more. "
|
63 |
-
"Language issues are derived from font-filtered text. "
|
64 |
-
"Regex issues and general document checks use unfiltered text from the original PDF. "
|
65 |
-
"All issue coordinates (if found) are mapped back to the original PDF."
|
66 |
-
)
|
67 |
-
|
68 |
with gr.Row():
|
69 |
file_input = gr.File(
|
70 |
label="Upload PDF",
|
71 |
-
file_types=[".pdf"]
|
72 |
-
#
|
73 |
-
|
74 |
)
|
75 |
|
76 |
with gr.Row():
|
77 |
-
analyze_btn = gr.Button("Analyze PDF")
|
78 |
|
79 |
with gr.Row():
|
80 |
results_output = gr.JSON(
|
81 |
-
label="Analysis Results",
|
82 |
show_label=True
|
83 |
)
|
84 |
|
85 |
-
with gr.Row():
|
86 |
pdf_output = gr.File(
|
87 |
label="Annotated PDF (Placeholder - View Coordinates in JSON)",
|
88 |
show_label=True,
|
89 |
-
interactive=False
|
90 |
)
|
91 |
|
92 |
analyze_btn.click(
|
93 |
-
fn=process_upload,
|
94 |
-
inputs=[file_input],
|
95 |
-
outputs=[results_output, pdf_output]
|
96 |
-
)
|
97 |
-
return interface
|
98 |
-
|
99 |
-
if __name__ == "__main__":
|
100 |
print("\n--- Launching Gradio Interface ---")
|
101 |
# config.set_java_home() is called when config.py is imported.
|
102 |
|
|
|
103 |
try:
|
|
|
104 |
lt_test = language_tool_python.LanguageTool('en-US')
|
105 |
lt_test.close()
|
106 |
print("App: LanguageTool initialized successfully for test.")
|
107 |
except Exception as lt_e:
|
108 |
-
print(f"App: Warning: Could not initialize LanguageTool for test. Language checks might fail: {lt_e}
|
109 |
print("Please ensure Java is installed and JAVA_HOME is correctly set (see config.py).")
|
110 |
|
111 |
app_interface = create_interface()
|
112 |
app_interface.launch(
|
113 |
-
share=False,
|
|
|
114 |
)
|
|
|
1 |
# app.py
|
2 |
import json
|
3 |
import traceback
|
4 |
+
import tempfile
|
5 |
import os
|
6 |
import gradio as gr
|
7 |
+
from typing import Tuple, Optional, Any # Added Any for file_data_binary
|
8 |
|
9 |
# Import config first to ensure JAVA_HOME is set early
|
10 |
import config
|
11 |
|
12 |
from main_analyzer import analyze_pdf
|
13 |
+
# Import language_tool_python only for the test in __main__ if needed
|
14 |
+
# import language_tool_python
|
15 |
+
|
16 |
+
def process_upload(file_data_binary: Optional[Any]) -> Tuple[str, Optional[str]]: # Use Optional[Any] for Gradio File type="binary"
|
17 |
+
if file_data_binary is None or not hasattr(file_data_binary, 'read'): # Check if it's a file-like object
|
18 |
+
# Gradio's binary type for gr.File returns a tempfile._TemporaryFileWrapper object
|
19 |
+
# If it's None, no file was uploaded.
|
20 |
+
# If it's not None but doesn't have 'read', it's an unexpected type.
|
21 |
+
# However, gradio usually passes the bytes directly if type="binary" was used in older versions
|
22 |
+
# or a TemporaryFileWrapper which is file-like.
|
23 |
+
# For robustness, let's check if it's bytes.
|
24 |
+
if isinstance(file_data_binary, bytes):
|
25 |
+
pass # Good, it's bytes
|
26 |
+
elif file_data_binary is None:
|
27 |
+
return json.dumps({"error": "No file uploaded or file data is None"}, indent=2), None
|
28 |
+
elif not hasattr(file_data_binary, 'read'): # It's not None, not bytes, not file-like
|
29 |
+
return json.dumps({"error": f"Unexpected file data type: {type(file_data_binary)}"}), None
|
30 |
+
# If it has 'read', it's a file-like object, proceed.
|
31 |
+
|
32 |
+
# analyze_pdf now handles stream-to-temp-file logic internally via original_pdf_access_path
|
33 |
+
# So we can pass the file_data_binary (which is a file-like object from Gradio) directly.
|
34 |
+
|
35 |
+
|
36 |
+
|
37 |
+
|
38 |
+
|
39 |
|
40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
|
42 |
try:
|
43 |
+
print(f"App: Processing uploaded file...")
|
44 |
+
# If file_data_binary is bytes, wrap it in BytesIO for file-like interface
|
45 |
+
# analyze_pdf expects a path or a file-like object with read() and seek()
|
|
|
46 |
|
47 |
+
# Gradio with type="binary" gives a tempfile._TemporaryFileWrapper.
|
48 |
+
# This object is already file-like and can be passed directly.
|
49 |
+
# No need to create another temp file here in app.py if main_analyzer handles it.
|
50 |
|
51 |
+
# analyze_pdf will create its own temp file if it receives a stream.
|
52 |
+
results_dict, _ = analyze_pdf(file_data_binary)
|
53 |
+
|
54 |
+
|
55 |
+
|
56 |
+
|
57 |
+
|
58 |
+
|
59 |
|
60 |
results_json = json.dumps(results_dict, indent=2, ensure_ascii=False)
|
61 |
return results_json, None
|
62 |
|
63 |
except Exception as e:
|
64 |
+
print(f"Error in process_upload: {e}\n{traceback.format_exc()}")
|
65 |
+
error_message = json.dumps({"error": str(e), "traceback": traceback.format_exc()}, indent=2)
|
66 |
+
|
67 |
+
|
68 |
+
|
69 |
+
|
70 |
+
|
71 |
+
|
72 |
return error_message, None
|
73 |
+
# No finally block needed here for temp file, as analyze_pdf handles its own if it creates one
|
74 |
+
# and Gradio handles the temp file wrapper it provides.
|
75 |
+
|
76 |
|
77 |
def create_interface():
|
78 |
with gr.Blocks(title="PDF Analyzer") as interface:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
79 |
with gr.Row():
|
80 |
file_input = gr.File(
|
81 |
label="Upload PDF",
|
82 |
+
file_types=[".pdf"],
|
83 |
+
# type="binary" # Gradio's File component returns a TemporaryFileWrapper which is file-like
|
84 |
+
|
85 |
)
|
86 |
|
87 |
with gr.Row():
|
|
|
88 |
|
89 |
with gr.Row():
|
90 |
results_output = gr.JSON(
|
91 |
+
label="Analysis Results", # Simplified label
|
92 |
show_label=True
|
93 |
)
|
94 |
|
|
|
95 |
pdf_output = gr.File(
|
96 |
label="Annotated PDF (Placeholder - View Coordinates in JSON)",
|
97 |
show_label=True,
|
98 |
+
interactive=False # Not interactive as it's a placeholder
|
99 |
)
|
100 |
|
101 |
analyze_btn.click(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
102 |
print("\n--- Launching Gradio Interface ---")
|
103 |
# config.set_java_home() is called when config.py is imported.
|
104 |
|
105 |
+
# Optional: Test LanguageTool initialization
|
106 |
try:
|
107 |
+
import language_tool_python # Import here for the test
|
108 |
lt_test = language_tool_python.LanguageTool('en-US')
|
109 |
lt_test.close()
|
110 |
print("App: LanguageTool initialized successfully for test.")
|
111 |
except Exception as lt_e:
|
112 |
+
print(f"App: Warning: Could not initialize LanguageTool for test. Language checks might fail: {lt_e}")
|
113 |
print("Please ensure Java is installed and JAVA_HOME is correctly set (see config.py).")
|
114 |
|
115 |
app_interface = create_interface()
|
116 |
app_interface.launch(
|
117 |
+
share=False, # Set to True for public link if ngrok is installed and desired
|
118 |
+
# server_port=7860 # Optionally specify a port
|
119 |
)
|