Spaces:
Sleeping
Sleeping
File size: 11,610 Bytes
334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 1a40b86 334c1a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 |
# app.py
import streamlit as st
import os
import sys
import logging
# Add the project root to the sys.path to allow importing modules like config, document_processing, etc.
# This assumes app.py is in the project root directory.
# Adjust the path if your app.py is in a subdirectory.
try:
# Import the necessary functions from your main script print("YES1")
from main import process_uploaded_files, setup_retrieval_system, summarize_extracted_documents
# Configure Streamlit's logging to match your application's settings
logging.basicConfig(level='INFO', format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
logger = logging.getLogger(__name__)
logger.info("Streamlit app started and logging configured.")
# Flag to check if modules were imported successfully
modules_loaded = True
except ImportError as e:
st.error(f"Could not import application modules. Please ensure your project structure is correct and dependencies are installed.")
st.error(f"ImportError: {e}")
logger = logging.getLogger(__name__)
logger.error(f"Failed to import application modules: {e}", exc_info=True)
modules_loaded = False # Set flag to False if imports fail
# --- Streamlit App Configuration ---
st.set_page_config(
page_title="Aya Insight Document Summarizer",
page_icon="📄",
layout="wide"
)
# --- Session State Initialization ---
# Initialize session state variables if they don't exist
if 'api_key_entered' not in st.session_state:
st.session_state.api_key_entered = False
if 'summary_results' not in st.session_state:
st.session_state.summary_results = None
if 'selected_filename' not in st.session_state:
st.session_state.selected_filename = None
# --- API Key Input Section ---
if not st.session_state.api_key_entered:
st.title("🔒 Enter Your Cohere API Key to Unlock")
api_key = st.text_input("Cohere API Key", type="password", help="Enter your Cohere API key to use the summarization service.")
if st.button("Unlock"):
if api_key:
# Basic validation: Just check if it's not empty.
# For a real application, you might want to validate by making a small API call.
os.environ["COHERE_API_KEY"] = api_key # Set the environment variable
st.session_state.api_key_entered = True
st.success("API Key accepted. You can now upload documents.")
st.rerun() # Rerun the app to show the main content
else:
st.warning("Please enter your Cohere API key.")
# --- Main Application Content (Unlocked) ---
if st.session_state.api_key_entered and modules_loaded:
st.title("📄 Aya Insight Document Summarizer")
st.markdown("""
Upload one or more PDF or image files to get a structured summary for each document.
""")
# --- File Uploader ---
uploaded_files = st.file_uploader(
"Choose Document Files",
type=["pdf", "png", "jpg", "jpeg", "tiff", "bmp", "gif"], # Added image types
accept_multiple_files=True,
help="You can upload multiple PDF or image documents here."
)
# --- Summarize Button and Logic ---
if uploaded_files: # Only show button if files are uploaded
st.info(f"You have uploaded {len(uploaded_files)} file(s).")
if st.button("Generate Summaries", key="summarize_button"):
st.session_state.selected_filename = None # Reset selected file on new summary generation
if not uploaded_files:
st.warning("Please upload at least one file before generating summaries.")
else:
st.subheader("Processing Documents...")
all_summary_results = [] # To store results for display
# Use a spinner to indicate processing
with st.spinner("Processing documents and generating summaries... This may take a few minutes depending on file size and number."):
try:
# Step 1: Process uploaded files (Extraction)
logger.info(f"Calling process_uploaded_files with {len(uploaded_files)} files.")
extraction_results = process_uploaded_files(uploaded_files)
logger.info(f"Finished document extraction. {len(extraction_results)} results obtained.")
# Check if any files were successfully extracted
if not any(res.get('text') for res in extraction_results):
st.error("No text could be extracted from the uploaded files. Please check the file formats.")
logger.error("No text extracted from any uploaded file.")
st.session_state.summary_results = [] # Store empty results
# st.stop() # Don't stop, allow user to try again
# Step 2: Setup retrieval system (Vector Store and Embedding)
logger.info("Calling setup_retrieval_system.")
extraction_results_with_chunks, retriever = setup_retrieval_system(extraction_results)
logger.info("Retriever system setup complete.")
# Step 3: Summarize the extracted documents
logger.info("Calling summarize_extracted_documents.")
summary_results = summarize_extracted_documents(extraction_results_with_chunks, retriever)
logger.info(f"Finished summarization. {len(summary_results)} summary results obtained.")
st.session_state.summary_results = summary_results # Store results in session state
except FileNotFoundError as fnf_error:
st.error(f"Configuration Error: {fnf_error}. Please check your environment settings.")
logger.error(f"Configuration Error during Streamlit process: {fnf_error}", exc_info=True)
st.session_state.summary_results = [] # Store empty results on error
except Exception as e:
st.error(f"An unexpected error occurred during processing: {e}")
logger.error(f"An unexpected error occurred during Streamlit process: {e}", exc_info=True)
st.session_state.summary_results = [] # Store empty results on error
# --- Display Document Tiles and Summaries ---
if st.session_state.summary_results is not None:
st.subheader("Summaries:")
if not st.session_state.summary_results:
st.info("No summaries were generated. Upload files and click 'Generate Summaries'.")
else:
# Display files as a grid of clickable tiles
files_per_row = 3
rows = len(st.session_state.summary_results) // files_per_row + (len(st.session_state.summary_results) % files_per_row > 0)
# Create a list of filenames for easy access
filenames = [res.get('filename', f'File {i+1}') for i, res in enumerate(st.session_state.summary_results)]
for i in range(rows):
cols = st.columns(files_per_row)
for j in range(files_per_row):
file_index = i * files_per_row + j
if file_index < len(st.session_state.summary_results):
result = st.session_state.summary_results[file_index]
filename = result.get('filename', f'File {file_index+1}')
is_selected = st.session_state.selected_filename == filename
# Create a tile using a button or markdown link
# Using a button inside a column for simplicity
with cols[j]:
# Add a border or highlight if selected
tile_style = "border: 2px solid lightgrey; padding: 10px; margin: 5px; text-align: center; cursor: pointer;"
if is_selected:
tile_style = "border: 2px solid steelblue; padding: 10px; margin: 5px; text-align: center; cursor: pointer; background-color: #e6f3ff;" # Highlight color
# Use markdown with HTML to create the clickable tile appearance
# When clicked, set the selected filename in session state
st.markdown(
f"""
<div style="{tile_style}" onclick="document.getElementById('hidden_button_{file_index}').click()">
📄<br>
<strong>{filename}</strong>
</div>
<button id="hidden_button_{file_index}" style="display: none;" onclick="document.getElementById('hidden_button_{file_index}').click()"></button>
""",
unsafe_allow_html=True
)
# Streamlit buttons don't work directly with markdown clicks like this easily.
# A simpler approach is to use a standard button and handle the click.
# Let's use a standard button instead of complex markdown/JS.
# Alternative using a standard button:
if st.button(f"📄 {filename}", key=f"tile_button_{file_index}"):
st.session_state.selected_filename = filename
logger.info(f"Selected file: {filename}")
st.rerun() # Rerun to display the summary
# Display summary of the selected file
if st.session_state.selected_filename:
st.markdown("---") # Separator
st.subheader(f"Summary for: {st.session_state.selected_filename}")
# Find the summary for the selected file
selected_summary = None
selected_result = None
for result in st.session_state.summary_results:
if result.get('filename') == st.session_state.selected_filename:
selected_summary = result.get('summary')
selected_result = result
break
if selected_summary:
if selected_result.get('success'):
st.markdown(selected_summary) # Render markdown summary
else:
st.error(f"Could not load summary for {st.session_state.selected_filename}: {selected_result.get('error', 'Unknown error')}")
else:
st.info(f"Summary not available for {st.session_state.selected_filename}.")
# Display overall processing status
successful_count = sum(res.get('success', False) for res in st.session_state.summary_results)
total_files = len(st.session_state.summary_results)
st.markdown(f"---") # Final separator
st.success(f"Processed {total_files} files. Successfully summarized {successful_count}.")
if successful_count < total_files:
st.warning("Some files could not be processed or summarized. See error messages above.")
# --- Message if API Key is not entered and modules loaded ---
if not st.session_state.api_key_entered:
st.info("Enter your Cohere API Key above to unlock the application functionality.")
|