Added all code

app.py

@@ -0,0 +1,97 @@
+import datetime as dt
+import pandas as pd
+
+import streamlit as st 
+from peopledatalabs import PDLPY as pdl
+
+
+def setup():
+    """
+    Streamlit related setup. This has to be run for each page.
+    """
+    hide_streamlit_style = """
+    
+    <style>
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
+    </style>
+    """
+    st.markdown(hide_streamlit_style, unsafe_allow_html=True)
+    
+def main():
+    '''
+     Main function of the app.
+    '''
+    setup()
+    st.title('People Data Labs Query Tool')
+    
+    st.subheader('Search for a Person or Company')
+    
+    st.write('''
+                This tool allows you to search for a person or company using the People Data Labs API. \n
+                *Guidelines*: \n
+                    - API Key: Your API key from People Data Labs \n
+                    - Type: Person or Company \n
+                    - SQL Query: The SQL query you want to run. Find it on the dashboard \n
+                    - Dataset: The dataset you want to search in. Leave blank for all   \n
+                    - Size: The number of results you want to return \n
+                    - Pretty Print: Whether you want the results to be pretty printed. Default is False \n \n\n
+            ''')
+
+    #horizontal line 
+    st.markdown('<hr>', unsafe_allow_html=True)
+    st.markdown('<br>', unsafe_allow_html=True)
+    
+    st.subheader('Search')
+    form = st.form(key='my_form')   
+    api_key = form.text_input(label='API Key')
+    type = form.selectbox(label='Type', options=['Person', 'Company'])
+    sql_query = form.text_area(label='SQL Query')
+    sql_query = sql_query.strip('"')
+    dataset = form.text_input(label='Dataset')
+    if dataset is None or dataset.strip() == '':
+        dataset = 'all'
+    size = form.number_input(label='Size', min_value=1, max_value=1000, value=10)
+    pretty = form.checkbox(label='Pretty Print')
+    submit_button = form.form_submit_button(label='Submit')
+    
+    if submit_button:
+        PARAMS = {'sql': sql_query, 'dataset': dataset, 'size': size, 'pretty': pretty}
+        
+        client = pdl(api_key=api_key)
+        
+        if type == 'Person':
+            response = client.person.search(**PARAMS).json()
+            
+        elif type == 'Company':
+            response = client.company.search(**PARAMS).json()
+            
+        if response["status"] == 200:
+            st.success(f"Found {response['total']} records for this search.")
+            data = response['data']
+            
+            #convert json to csv. Json is list of dicts
+            df = pd.DataFrame(data)
+            csv = df.to_csv(index=False)
+            
+            #show data
+            st.markdown('<hr>', unsafe_allow_html=True)
+            st.markdown('<br>', unsafe_allow_html=True)
+            st.subheader('Data')
+            st.download_button(label='Download CSV', data=csv, file_name=f'{type}-data-{dt.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}.csv')
+            st.dataframe(df)
+        else:
+            st.error('Error retrieving data!')
+            st.error(f"Error: {response}")
+            
+if __name__ == '__main__':
+    main()
+            
+            
+        
+    
+
+
+    
+
+    

requirements.txt

@@ -0,0 +1,3 @@
+pandas==2.0.3
+peopledatalabs==1.1.9
+streamlit==1.24.1