sid22669 commited on
Commit
b7ec1f2
·
verified ·
1 Parent(s): bb41517

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -3
app.py CHANGED
@@ -88,11 +88,25 @@ uploaded_file = st.file_uploader(
88
  print("Somethinguploaded")
89
 
90
  if uploaded_file:
91
- # Check if serial_id already exists in session for current file, else create
92
- if "uploaded_file_name" not in st.session_state or st.session_state.uploaded_file_name != uploaded_file.name:
 
 
 
 
 
 
 
 
 
 
 
 
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 # To store correction during session
 
 
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: