import gradio as gr from df.PaperCentral import PaperCentral from gradio_calendar import Calendar from datetime import datetime, timedelta from typing import Union, List from author_leaderboard_tab import author_leaderboard_tab # Initialize the PaperCentral class instance paper_central_df = PaperCentral() # Create the Gradio Blocks app with custom CSS with gr.Blocks(css="style.css") as demo: gr.Markdown("# Paper Central") with gr.Tab("Paper-central"): with gr.Accordion(label="⭐Release notes", open=False): gr.Markdown(""" - 8/10/2024 - MICCAI proceedings added - 7/10/2024 - COLM2024 proceedings added - 4/10/2024 - You can now filter by Title """) # Create a row for navigation buttons and calendar with gr.Row(): with gr.Column(scale=1): # Define the 'Next Day' and 'Previous Day' buttons next_day_btn = gr.Button("Next Day") prev_day_btn = gr.Button("Previous Day") with gr.Column(scale=4): # Define the calendar component for date selection calendar = Calendar( type="datetime", label="Select a date", info="Click the calendar icon to bring up the calendar.", value=datetime.today().strftime('%Y-%m-%d') # Default to today's date ) # Create a row for Hugging Face options and Conference options with gr.Row(): with gr.Column(): # Define the checkbox group for Hugging Face options cat_options = gr.CheckboxGroup( label="Category", choices=[ 'cs.*', 'eess.*', 'econ.*', 'math.*', 'astro-ph.*', 'cond-mat.*', 'gr-qc', 'hep-ex', 'hep-lat', 'hep-ph', 'hep-th', 'math-ph', 'nlin.*', 'nucl-ex', 'nucl-th', 'physics.*', 'quant-ph', 'q-bio.*', 'q-fin.*', 'stat.*', ], value=["cs.*"] ) hf_options = gr.CheckboxGroup( label="Hugging Face options", choices=["🤗 artifacts", "datasets", "models", "spaces", "github"], value=[], elem_id="hf_options" ) with gr.Column(): # Define the checkbox group for Conference options conference_options = gr.CheckboxGroup( label="Conference options", choices=["In proceedings"] + PaperCentral.CONFERENCES ) with gr.Row(): # Define a Textbox for author search author_search = gr.Textbox( label="Search Authors", placeholder="Enter author name", ) title_search = gr.Textbox( label="Search Title", placeholder="Enter keywords", ) # Define the Dataframe component to display paper data # List of columns in your DataFrame columns = paper_central_df.COLUMNS_START_PAPER_PAGE paper_central_component = gr.Dataframe( label="Paper Data", value=paper_central_df.df_prettified[columns], datatype=[ paper_central_df.DATATYPES[column] for column in columns ], row_count=(0, "dynamic"), interactive=False, max_height=1000, elem_id="table", wrap=True, ) with gr.Tab("Leaderboard"): author_leaderboard_tab() # Define function to move to the next day def go_to_next_day( date: Union[str, datetime], cat_options_list: List[str], hf_options_list: List[str], conference_options_list: List[str], author_search_input: str, title_search_input: str, ) -> tuple: """ Moves the selected date to the next day and updates the data. Args: date (Union[str, datetime]): The current date selected in the calendar. cat_options_list (List[str]): List of selected Category options. hf_options_list (List[str]): List of selected Hugging Face options. conference_options_list (List[str]): List of selected Conference options. Returns: tuple: The new date as a string and the updated Dataframe component. """ # Ensure the date is in string format if isinstance(date, datetime): date_str = date.strftime('%Y-%m-%d') else: date_str = date # Parse the date string and add one day new_date = datetime.strptime(date_str, '%Y-%m-%d') + timedelta(days=1) new_date_str = new_date.strftime('%Y-%m-%d') # Update the Dataframe updated_data = paper_central_df.filter( selected_date=new_date_str, cat_options=cat_options_list, hf_options=hf_options_list, conference_options=conference_options_list, author_search_input=author_search_input, title_search_input=title_search_input, ) # Return the new date and updated Dataframe return new_date_str, updated_data # Define function to move to the previous day def go_to_previous_day( date: Union[str, datetime], cat_options_list: List[str], hf_options_list: List[str], conference_options_list: List[str], author_search_input: str, title_search_input: str, ) -> tuple: """ Moves the selected date to the previous day and updates the data. Args: date (Union[str, datetime]): The current date selected in the calendar. cat_options_list (List[str]): List of selected Category options. hf_options_list (List[str]): List of selected Hugging Face options. conference_options_list (List[str]): List of selected Conference options. Returns: tuple: The new date as a string and the updated Dataframe component. """ # Ensure the date is in string format if isinstance(date, datetime): date_str = date.strftime('%Y-%m-%d') else: date_str = date # Parse the date string and subtract one day new_date = datetime.strptime(date_str, '%Y-%m-%d') - timedelta(days=1) new_date_str = new_date.strftime('%Y-%m-%d') # Update the Dataframe updated_data = paper_central_df.filter( selected_date=new_date_str, cat_options=cat_options_list, hf_options=hf_options_list, conference_options=conference_options_list, author_search_input=author_search_input, title_search_input=title_search_input, ) # Return the new date and updated Dataframe return new_date_str, updated_data # Define function to update data when date or options change def update_data( date: Union[str, datetime], cat_options_list: List[str], hf_options_list: List[str], conference_options_list: List[str], author_search_input: str, title_search_input: str, ): """ Updates the data displayed in the Dataframe based on the selected date and options. Args: date (Union[str, datetime]): The selected date. cat_options_list (List[str]): List of selected Category options. hf_options_list (List[str]): List of selected Hugging Face options. conference_options_list (List[str]): List of selected Conference options. Returns: gr.Dataframe.update: An update object for the Dataframe component. """ return paper_central_df.filter( selected_date=date, cat_options=cat_options_list, hf_options=hf_options_list, conference_options=conference_options_list, author_search_input=author_search_input, title_search_input=title_search_input, ) # Function to handle conference options change def on_conference_options_change( date: Union[str, datetime], cat_options_list: List[str], hf_options_list: List[str], conference_options_list: List[str], author_search_input: str, title_search_input: str, ): cat_options_update = gr.update() paper_central_component_update = gr.update() visible = True # Some conference options are selected # Update cat_options to empty list if conference_options_list: cat_options_update = gr.update(value=[]) paper_central_component_update = update_data( date, [], hf_options_list, conference_options_list, author_search_input, title_search_input, ) visible = False calendar_update = gr.update(visible=visible) next_day_btn_update = gr.update(visible=visible) prev_day_btn_update = gr.update(visible=visible) return paper_central_component_update, cat_options_update, calendar_update, next_day_btn_update, prev_day_btn_update # Function to handle category options change def on_cat_options_change( date: Union[str, datetime], cat_options_list: List[str], hf_options_list: List[str], conference_options_list: List[str], author_search_input: str, title_search_input: str, ): conference_options_update = gr.update() paper_central_component_update = gr.update() visible = False # Some category options are selected # Update conference_options to empty list if cat_options_list: conference_options_update = gr.update(value=[]) paper_central_component_update = update_data( date, cat_options_list, hf_options_list, [], author_search_input, title_search_input, ) visible = True calendar_update = gr.update(visible=visible) next_day_btn_update = gr.update(visible=visible) prev_day_btn_update = gr.update(visible=visible) return paper_central_component_update, conference_options_update, calendar_update, next_day_btn_update, prev_day_btn_update inputs = [ calendar, cat_options, hf_options, conference_options, author_search, title_search, ] # Set up the event listener for the author search author_search.submit( fn=update_data, inputs=inputs, outputs=paper_central_component, ) title_search.submit( fn=update_data, inputs=inputs, outputs=paper_central_component, ) # Set up the event listener for the 'Next Day' button next_day_btn.click( fn=go_to_next_day, inputs=inputs, outputs=[calendar, paper_central_component], ) # Set up the event listener for the 'Previous Day' button prev_day_btn.click( fn=go_to_previous_day, inputs=inputs, outputs=[calendar, paper_central_component], ) # Set up the event listener for the calendar date change calendar.change( fn=update_data, inputs=inputs, outputs=paper_central_component, ) # Set up the event listener for the Hugging Face options change hf_options.change( fn=update_data, inputs=inputs, outputs=paper_central_component, ) # Event chaining for conference options change conference_options.change( fn=on_conference_options_change, inputs=inputs, outputs=[paper_central_component, cat_options, calendar, next_day_btn, prev_day_btn], ) # Event chaining for category options change cat_options.change( fn=on_cat_options_change, inputs=inputs, outputs=[paper_central_component, conference_options, calendar, next_day_btn, prev_day_btn], ) # Load the initial data when the app starts request = gr.Request() def echo(request: gr.Request): """ Echoes the input text and prints request details. Args: text (str): The input text from the user. request (gr.Request): The HTTP request object. Returns: str: The echoed text. """ calendar = gr.update() conferences = gr.update() hf_options = gr.update() if request: print("Request headers dictionary:", dict(request.headers)) print("IP address:", request.client.host) print("Query parameters:", dict(request.query_params)) print("Session hash:", request.session_hash) if 'date' in request.query_params: calendar = gr.update(value=request.query_params['date']) if 'conferences' in request.query_params: conferences = request.query_params['conferences'] try: conferences = [value for value in conferences.split(',')] except ValueError: conferences = [] print(conferences) conferences = gr.update(value=conferences) if "hf_options" in request.query_params: hf_options = request.query_params['hf_options'] try: hf_options = [value for value in hf_options.split(',')] except ValueError: hf_options = [] hf_options = gr.update(value=hf_options) return calendar, conferences, hf_options demo.load( fn=update_data, inputs=inputs, outputs=paper_central_component, api_name=False, ).then( fn=echo, outputs=[calendar, conference_options, hf_options] ) # Define the main function to launch the app def main(): """ Launches the Gradio app. """ demo.launch() # Run the main function when the script is executed if __name__ == "__main__": main()