import streamlit as st
import requests
import pandas as pd

# Slider to select a value
x = st.slider('Select a value')
st.write(x, 'squared is', x * x)

# Function to get data from Hugging Face datasets server
def get_arxiv_data():
    url = "https://datasets-server.huggingface.co/first-rows?dataset=arxiv_dataset&config=default&split=train"
    response = requests.get(url)
    if response.status_code == 200:
        return response.json()
    else:
        st.error("Failed to retrieve data")
        return None

# Button to load data
if st.button('Load ArXiv Data'):
    data = get_arxiv_data()
    if data:
        # Assuming the data is in a format that can be directly converted to a DataFrame
        df = pd.DataFrame(data)
        st.write(df)

# import streamlit as st

# x = st.slider('Select a value')
# st.write(x, 'squared is', x * x)

# from datasets import load_dataset

# dataset = load_dataset("arxiv_dataset", data_dir="")

# print(dataset)