Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,11 +88,25 @@ uploaded_file = st.file_uploader(
|
|
| 88 |
print("Somethinguploaded")
|
| 89 |
|
| 90 |
if uploaded_file:
|
| 91 |
-
#
|
| 92 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
st.session_state.uploaded_file_name = uploaded_file.name
|
| 94 |
st.session_state.serial_id = str(uuid.uuid4())
|
| 95 |
-
st.session_state.corrected_prediction = None #
|
|
|
|
|
|
|
| 96 |
|
| 97 |
# Save uploaded file to temp and extract text (same as your code)
|
| 98 |
with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as temp_file:
|
|
|
|
| 88 |
print("Somethinguploaded")
|
| 89 |
|
| 90 |
if uploaded_file:
|
| 91 |
+
# Ensure /tmp directory exists
|
| 92 |
+
upload_dir = "/tmp/uploads"
|
| 93 |
+
os.makedirs(upload_dir, exist_ok=True)
|
| 94 |
+
|
| 95 |
+
# Save the uploaded file to /tmp
|
| 96 |
+
file_path = os.path.join(upload_dir, uploaded_file.name)
|
| 97 |
+
with open(file_path, "wb") as f:
|
| 98 |
+
f.write(uploaded_file.getbuffer())
|
| 99 |
+
|
| 100 |
+
# Handle session state tracking
|
| 101 |
+
if (
|
| 102 |
+
"uploaded_file_name" not in st.session_state
|
| 103 |
+
or st.session_state.uploaded_file_name != uploaded_file.name
|
| 104 |
+
):
|
| 105 |
st.session_state.uploaded_file_name = uploaded_file.name
|
| 106 |
st.session_state.serial_id = str(uuid.uuid4())
|
| 107 |
+
st.session_state.corrected_prediction = None # Reset correction state
|
| 108 |
+
|
| 109 |
+
st.success(f"File uploaded and saved to: {file_path}")
|
| 110 |
|
| 111 |
# Save uploaded file to temp and extract text (same as your code)
|
| 112 |
with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as temp_file:
|