import streamlit as st
import os 
import subprocess
import geocoder
import pandas as pd 
from csv import writer
# import keyboard
# st.title('Dawg Detection 🐶')
st.markdown("<h1 style='text-align: center; color: white;'>Dawg Detection 🐶</h1>", unsafe_allow_html=True)
jury=True
if 'jury' not in st.session_state:
    st.session_state.jury = False

butt=st.button('Configure',disabled=st.session_state.jury,help='Press This And Wait A Few Seconds If You Are Running The Website For The First Time')
if butt==True:
    os.system('git clone https://github.com/WongKinYiu/yolov7')
    os.system('%cd yolov7')
    os.system('pip install -r requirements.txt')
    os.system('ls')
    os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt')
    os.system('pip install opencv_python')
    os.system('conda install pytorch torchvision torchaudio -c pytorch -y')
    os.system('pip install pyyaml')
    os.system('pip install scipy')
    st.session_state.jury = True


bob='python yolov7/detect.py --weights yolov7.pt --classes 16 --conf 0.25 --img-size 640 --source  Images/seesh.png  --exist-ok --save-conf --save-txt'
tusk='python yolov7/detect.py --weights yolov7.pt --classes 16 --conf 0.25 --img-size 640 --source Images/seesh.png --exist-ok --save-conf --save-txt'


pic=st.camera_input('Take A Picture')
if pic:
    with open ('Images/seesh.png','wb') as file:
        file.write(pic.getbuffer())
        os.system(bob)
        st.text('Detected Dawgs')
        img=st.image('runs/detect/exp/seesh.png')

        #count
        f=open("runs/detect/exp/labels/seesh.txt",'r')
        a=f.readlines()
        b=[]
        for j in range(0,len(a)):
            if a[j][0:2] == '16':
                b.append('yass')
            else:
                continue 
        print(len(b))
        f=open("runs/detect/exp/labels/seesh.txt",'w')
        st.balloons()
        st.success(f'Total Number Of Dogs Detected: {len(b)}')
        st.markdown("---")

        #map
        g = geocoder.ip('me')
        print(g.latlng)
        with open('database/cord.csv', 'a') as f_object:
            writer_object = writer(f_object)
            writer_object.writerow(g.latlng)
            f_object.close()
        df=pd.read_csv('database/cord.csv')
        cum=len(df['lat'])
        st.text(f'Your Location Coordinates: {g.latlng}')
        st.map(df,use_container_width=True,zoom = 4)
        st.markdown(f"<h5 style='color: red;'>Total Number Of Reports Around The World: {cum}<h5>", unsafe_allow_html=True)

clicked= st.button('Click Me For Balloons',help='Click here to feel happy 😁')
if clicked==True:
    st.balloons()