Update app.py
Browse files
app.py
CHANGED
|
@@ -183,33 +183,6 @@ def format_prompt_inputs(image_collection, text_collection, video_collection, us
|
|
| 183 |
|
| 184 |
return inputs
|
| 185 |
|
| 186 |
-
def unzip_file(zip_path, extract_to):
|
| 187 |
-
"""
|
| 188 |
-
Unzips a zip file to the specified directory.
|
| 189 |
-
|
| 190 |
-
Args:
|
| 191 |
-
zip_path (str): Path to the zip file.
|
| 192 |
-
extract_to (str): Directory where the contents should be extracted.
|
| 193 |
-
"""
|
| 194 |
-
try:
|
| 195 |
-
# Ensure the destination directory exists
|
| 196 |
-
os.makedirs(extract_to, exist_ok=True)
|
| 197 |
-
|
| 198 |
-
# Open the zip file
|
| 199 |
-
with zipfile.ZipFile(zip_path, 'r') as zip_ref:
|
| 200 |
-
# Extract all the contents
|
| 201 |
-
zip_ref.extractall(extract_to)
|
| 202 |
-
return True
|
| 203 |
-
except Exception as e:
|
| 204 |
-
print(f"An error occurred: {e}")
|
| 205 |
-
return False
|
| 206 |
-
|
| 207 |
-
def process_pdf(pdf_path):
|
| 208 |
-
# Placeholder function to simulate PDF processing
|
| 209 |
-
# Replace this with actual PDF processing logic, such as extracting text, images, etc.
|
| 210 |
-
time.sleep(2) # Simulating processing delay
|
| 211 |
-
return "image_collection", "text_collection", "video_collection" # Replace with actual collections
|
| 212 |
-
|
| 213 |
def page_1():
|
| 214 |
st.title("Page 1: Upload and Process PDFs")
|
| 215 |
|
|
@@ -229,7 +202,6 @@ def page_1():
|
|
| 229 |
try:
|
| 230 |
total_files = len(uploaded_pdf_files)
|
| 231 |
files_processed = 0
|
| 232 |
-
progress_step = 100 / total_files if total_files > 0 else 0
|
| 233 |
|
| 234 |
# Process PDF files
|
| 235 |
for uploaded_pdf in uploaded_pdf_files:
|
|
@@ -239,7 +211,7 @@ def page_1():
|
|
| 239 |
|
| 240 |
# Simulate PDF processing (replace with actual PDF processing logic)
|
| 241 |
files_processed += 1
|
| 242 |
-
progress_bar.progress(files_processed
|
| 243 |
status_text.text(f"Processing PDF: {uploaded_pdf.name} ({files_processed}/{total_files})")
|
| 244 |
|
| 245 |
# Call your actual PDF processing function here
|
|
|
|
| 183 |
|
| 184 |
return inputs
|
| 185 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 186 |
def page_1():
|
| 187 |
st.title("Page 1: Upload and Process PDFs")
|
| 188 |
|
|
|
|
| 202 |
try:
|
| 203 |
total_files = len(uploaded_pdf_files)
|
| 204 |
files_processed = 0
|
|
|
|
| 205 |
|
| 206 |
# Process PDF files
|
| 207 |
for uploaded_pdf in uploaded_pdf_files:
|
|
|
|
| 211 |
|
| 212 |
# Simulate PDF processing (replace with actual PDF processing logic)
|
| 213 |
files_processed += 1
|
| 214 |
+
progress_bar.progress(files_processed / total_files) # Updated progress calculation
|
| 215 |
status_text.text(f"Processing PDF: {uploaded_pdf.name} ({files_processed}/{total_files})")
|
| 216 |
|
| 217 |
# Call your actual PDF processing function here
|