Update app.py

app.py

@@ -1,8 +1,9 @@
 import streamlit as st
-import requests
-
-# Define your FastAPI backend URL
-BACKEND_URL = "http://127.0.0.2:8001"
+import os
+import sqlite3
+from home import show_home_page
+from auth import signup, login
+import database  # Ensure database initialization
 
 def main():
     st.set_page_config(page_title="Auto Assist Chatbot", layout="wide")
@@ -50,6 +51,19 @@ def main():
         unsafe_allow_html=True
     )
 
+    # Function to create a connection to the SQLite database
+    def create_connection():
+        db_path = os.path.join(os.path.dirname(__file__), 'database.db')
+        conn = sqlite3.connect(db_path)
+        return conn
+
+    def is_valid_input(username, password):
+        prefix = "autoassist"
+        return username.startswith(prefix) and password.startswith(prefix)
+
+    # Ensure table creation
+    database.create_table()
+
     if st.session_state.logged_in:
         if st.sidebar.button('Back to Login'):
             st.session_state.logged_in = False
@@ -67,11 +81,8 @@ def main():
             if st.button('Signup'):
                 if password_signup == confirm_password_signup:
                     if is_valid_input(username_signup, password_signup):
-                        response = requests.post(f"{BACKEND_URL}/signup/", json={"username": username_signup, "password": password_signup})
-                        if response.status_code == 200:
-                            st.success('Signup successful!')
-                        else:
-                            st.error('Signup failed. Please try again.')
+                        signup(username_signup, password_signup)
+                        st.success('Signup successful!')
                     else:
                         st.error('You are not able to use "autoassist chatbot".')
                 else:
@@ -84,8 +95,7 @@ def main():
 
             if st.button('Login'):
                 if is_valid_input(username_login, password_login):
-                    response = requests.post(f"{BACKEND_URL}/login/", json={"username": username_login, "password": password_login})
-                    if response.status_code == 200:
+                    if login(username_login, password_login):
                         st.session_state.logged_in = True
                         st.experimental_rerun()
                     else: