Spaces:
Build error
Build error
| import streamlit as st | |
| import pandas as pd | |
| import numpy as np | |
| from src.Surveyor import Surveyor | |
| def run_survey(surveyor, research_keywords, max_search, num_papers): | |
| zip_file_name, survey_file_name = surveyor.survey(research_keywords, | |
| max_search=max_search, | |
| num_papers=num_papers | |
| ) | |
| with open(str(zip_file_name), "rb") as file: | |
| btn = st.download_button( | |
| label="Download extracted topic-clustered-highlights, images and tables as zip", | |
| data=file, | |
| file_name=str(zip_file_name) | |
| ) | |
| with open(str(survey_file_name), "rb") as file: | |
| btn = st.download_button( | |
| label="Download detailed generated survey file", | |
| data=file, | |
| file_name=str(survey_file_name) | |
| ) | |
| for line in file.readlines(): | |
| st.write(line) | |
| def survey_space(surveyor): | |
| form = st.sidebar.form(key='survey_form') | |
| research_keywords = form.text_input("What would you like to research in today?") | |
| max_search = form.number_input("num_papers_to_search", help="maximium number of papers to glance through - defaults to 20", | |
| min_value=1, max_value=60, value=10, step=1, key='max_search') | |
| num_papers = form.number_input("num_papers_to_select", help="maximium number of papers to select and analyse - defaults to 8", | |
| min_value=1, max_value=25, value=2, step=1, key='num_papers') | |
| submit = form.form_submit_button('Submit') | |
| if submit: | |
| run_survey(surveyor, research_keywords, max_search, num_papers) | |
| if __name__ == '__main__': | |
| st.title('Auto-Research V0.1 - Automated Survey generation from research keywords') | |
| std_col, survey_col = st.columns(2) | |
| std_col.header('execution log:') | |
| survey_col.header('Generated_survey:') | |
| with st.spinner('Loading The-Surveyor ...'): | |
| surveyor_obj = Surveyor(print_fn=std_col.write, survey_print_fn=survey_col.write, refresh_models=True) | |
| survey_space(surveyor_obj) | |