app.py

import gradio as gr
from arxiv_metadata_service import ArxivMetadataService
import traceback

arxiv_service = ArxivMetadataService()

def extract_metadata(query: str, max_results: int):
    try:
        return arxiv_service.extract_and_update(query, max_results)
    except Exception as e:
        error_msg = f"An error occurred: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
        return error_msg

demo = gr.Interface(
    fn=extract_metadata,
    inputs=[
        gr.Textbox(label="ArXiv Query"),
        gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Max Results")
    ],
    outputs="text",
    title="ArXiv Metadata Extractor",
    description="Extract metadata from ArXiv papers and update the dataset."
)

if __name__ == "__main__":
    demo.launch()