Update app.py

app.py

@@ -58,39 +58,55 @@ def convert_to_excel(data):
     df.to_excel(writer, sheet_name='Grants', index=False)
   return buffer.getvalue()
 
-st.title("Quantilytix Grant Scraper")
-
-url = st.text_input("Enter URL")
-
-if "scraped_data" not in st.session_state:
-    st.session_state.scraped_data = None
-
-if st.button("Get grants"):
-    if url:
-        try:
-            with st.spinner("Retrieving Grants, Please Wait...."):
-                result = get_data(url)
-                st.session_state.scraped_data = result  # Store result in session state
-                st.success("Data scraped successfully!")
-        except Exception as e:
-            st.error(f"Error scraping data: {e}")
-    else:
-        st.warning("Please enter a URL.")
-
-if st.session_state.scraped_data:
-    selected_format = st.selectbox("Select Download Format", ("CSV", "Excel"))
-
-    result = st.session_state.scraped_data  # Access the saved result
-
-    if selected_format == "CSV":
-        csv_data = convert_to_csv(result)
-        b64 = base64.b64encode(csv_data).decode()
-        download_link = f"<a href='data:application/vnd.ms-excel;base64,{b64}' download='grants.csv'>Download CSV</a>"
-        st.markdown(download_link, unsafe_allow_html=True)
-    elif selected_format == "Excel":
-        excel_data = convert_to_excel(result)
-        b64 = base64.b64encode(excel_data).decode()
-        download_link = f"<a href='data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64}' download='grants.xlsx'>Download Excel</a>"
-        st.markdown(download_link, unsafe_allow_html=True)
-
-    st.dataframe(result['grants'])
+
+def main():
+    st.sidebar.title("Quantilytix Grant Scraper")
+
+    url = st.sidebar.text_input("Enter URL")
+
+    if "scraped_data" not in st.session_state:
+        st.session_state.scraped_data = None
+
+    if st.sidebar.button("Get grants"):
+        if url:
+            try:
+                with st.spinner("Retrieving Grants, Please Wait...."):
+                    result = get_data(url)
+                    st.session_state.scraped_data = result  # Store result in session state
+                    st.success("Data scraped successfully!")
+            except Exception as e:
+                st.error(f"Error scraping data: {e}")
+        else:
+            st.warning("Please enter a URL.")
+
+    if st.session_state.scraped_data:
+        selected_format = st.sidebar.selectbox("Select Download Format", ("CSV", "Excel"))
+
+        result = st.session_state.scraped_data  # Access the saved result
+
+        if selected_format == "CSV":
+            csv_data = convert_to_csv(result)
+            b64 = base64.b64encode(csv_data).decode()
+            download_link = f"<a href='data:application/vnd.ms-excel;base64,{b64}' download='grants.csv'>Download CSV</a>"
+            st.sidebar.markdown(download_link, unsafe_allow_html=True)
+        elif selected_format == "Excel":
+            excel_data = convert_to_excel(result)
+            b64 = base64.b64encode(excel_data).decode()
+            download_link = f"<a href='data:application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;base64,{b64}' download='grants.xlsx'>Download Excel</a>"
+            st.sidebar.markdown(download_link, unsafe_allow_html=True)
+
+        st.dataframe(result['grants'])
+
+    if st.sidebar.button("Load as Knowledge Base"):
+        if st.session_state.scraped_data is not None:
+            st.write("Chat Interface Loaded. Start asking questions about the grants!")
+            query = st.text_input("Ask a question about the grants:", key="chat_input")
+            if query:
+                response = "This is where the response based on the knowledge base would be generated."  # Placeholder
+                st.write("Response:")
+                st.write(response)
+        else:
+            st.warning("No data available. Please scrape grants first.")
+
+if __name__ == "__main__":
+    main()