Spaces:
Sleeping
Sleeping
import streamlit as st | |
import pandas as pd | |
from huggingface_hub import Repository | |
import os | |
from pathlib import Path | |
import csv | |
# assign directory | |
auth_token = os.environ.get("space_to_dataset") or True | |
#data = pd.read_csv("test.csv") | |
#for line in data: | |
# st.write(line) | |
title = st.text_input('Movie title', 'Life of Brian') | |
if st.button('Submit'): | |
new_row = title | |
# data = data.append(new_row, ignore_index=True) | |
# st.write(data) | |
# st.write(os.getcwd()) | |
# data.to_csv('test.csv', index= False) | |
DATASET_REPO_URL = 'ppsingh/annotation_data' | |
DATA_FILENAME = "test.csv" | |
DATA_FILE = os.path.join("data", DATA_FILENAME) | |
repo = Repository( local_dir="data", clone_from=DATASET_REPO_URL, repo_type="dataset", use_auth_token= auth_token) | |
df = pd.read_csv('data/test.csv') | |
st.write(df) | |
# st.write('data/test.csv') | |
# iterate over files in | |
# that directory | |
directory = os.getcwd() | |
files = Path(directory).glob('*') | |
for file in files: | |
st.write(file) | |
with open(DATA_FILE, "a") as csvfile: | |
writer = csv.DictWriter(csvfile, fieldnames=["Sentences"]) | |
writer.writerow({'Sentences': new_row}) | |
repo.push_to_hub('adding new line') | |
st.write('Succcess') | |