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')