Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import gspread | |
| from google.oauth2.service_account import Credentials | |
| import pandas as pd | |
| def CreatePage(): | |
| st.header("View & Download Results") | |
| if "results" in st.session_state and st.session_state["results"]: | |
| results_df = pd.DataFrame(st.session_state["results"]) | |
| st.write("### Results Preview") | |
| # Display the results preview | |
| if "Extracted Information" in results_df.columns and "Search Results" in results_df.columns: | |
| st.dataframe(results_df.style.map(lambda val: 'background-color: #d3f4ff' if isinstance(val, str) else '', subset=["Extracted Information", "Search Results"])) | |
| else: | |
| st.warning("Required columns are missing in results data.") | |
| # Download options | |
| download_option = st.selectbox( | |
| "Select data to download:", | |
| ["All Results", "Extracted Information", "Web Results"] | |
| ) | |
| if download_option == "All Results": | |
| data_to_download = results_df | |
| elif download_option == "Extracted Information": | |
| data_to_download = results_df[["Entity", "Extracted Information"]] | |
| elif download_option == "Web Results": | |
| data_to_download = results_df[["Entity", "Search Results"]] | |
| st.download_button( | |
| label=f"Download {download_option} as CSV", | |
| data=data_to_download.to_csv(index=False), | |
| file_name=f"{download_option.lower().replace(' ', '_')}.csv", | |
| mime="text/csv" | |
| ) | |
| # Option to update Google Sheets | |
| update_option = st.selectbox( | |
| "Do you want to update Google Sheets?", | |
| ["No", "Yes"] | |
| ) | |
| if update_option == "Yes": | |
| if 'sheet_id' not in st.session_state: | |
| st.session_state.sheet_id = '' | |
| if 'range_name' not in st.session_state: | |
| st.session_state.range_name = '' | |
| # Input fields for Google Sheets ID and Range | |
| sheet_id = st.text_input("Enter Google Sheet ID", value=st.session_state.sheet_id) | |
| range_name = st.text_input("Enter Range (e.g., 'Sheet1!A1')", value=st.session_state.range_name) | |
| if sheet_id and range_name: | |
| st.session_state.sheet_id = sheet_id | |
| st.session_state.range_name = range_name | |
| # Prepare data for update | |
| data_to_update = [results_df.columns.tolist()] + results_df.values.tolist() | |
| # Update Google Sheets button | |
| if st.button("Update Google Sheet"): | |
| try: | |
| if '!' not in range_name: | |
| st.error("Invalid range format. Please use the format 'SheetName!Range'.") | |
| else: | |
| sheet_name, cell_range = range_name.split('!', 1) | |
| scopes = ["https://www.googleapis.com/auth/spreadsheets"] | |
| creds = Credentials.from_service_account_file("/Users/sam22ridhi/Desktop/data/DataScribe/credentials/credentials.json", scopes=scopes) | |
| client = gspread.authorize(creds) | |
| sheet = client.open_by_key(sheet_id).worksheet(sheet_name) | |
| sheet.clear() | |
| sheet.update(f"{cell_range}", data_to_update) | |
| st.success("Data updated in the Google Sheet!") | |
| except Exception as e: | |
| st.error(f"Error updating Google Sheet: {e}") | |
| else: | |
| st.warning("Please enter both the Sheet ID and Range name before updating.") | |
| else: | |
| st.warning("No results available to view. Please run the extraction process.") |