Update app.py

app.py

@@ -1,34 +1,49 @@
 import streamlit as st
 import psycopg2
 import os
+from dotenv import load_dotenv
 from datetime import datetime
 
+# Load environment variables from .env file
+load_dotenv()
+
+# Connect to PostgreSQL database
 def connect_db():
-    return psycopg2.connect(
-        host=os.getenv("DB_HOST"),
-        port=os.getenv("DB_PORT", 5432),
-        database=os.getenv("DB_NAME"),
-        user=os.getenv("DB_USER"),
-        password=os.getenv("DB_PASSWORD")
-    )
+    try:
+        return psycopg2.connect(
+            host=os.getenv("DB_HOST"),
+            port=os.getenv("DB_PORT", 5432),
+            database=os.getenv("DB_NAME"),
+            user=os.getenv("DB_USER"),
+            password=os.getenv("DB_PASSWORD")
+        )
+    except psycopg2.OperationalError as e:
+        st.error(f"Error connecting to the database: {e}")
+        return None
 
+# Insert user data into the "users" table
 def insert_user(company_name, requirements, date, status, notes, next_steps, contact_person, contact_information):
     conn = connect_db()
-    cur = conn.cursor()
-
-    query = """
-    INSERT INTO users 
-    (company_name, requirements, date, status, notes, next_steps, contact_person, contact_information)
-    VALUES (%s, %s, %s, %s, %s, %s, %s, %s)
-    """
+    if conn is None:
+        return
     
-    cur.execute(query, (company_name, requirements, date, status, notes, next_steps, contact_person, contact_information))
-    conn.commit()
-    cur.close()
-    conn.close()
-
-    st.success("User added successfully!")
+    try:
+        cur = conn.cursor()
+        query = """
+        INSERT INTO users 
+        (company_name, requirements, date, status, notes, next_steps, contact_person, contact_information)
+        VALUES (%s, %s, %s, %s, %s, %s, %s, %s)
+        """
+        cur.execute(query, (company_name, requirements, date, status, notes, next_steps, contact_person, contact_information))
+        conn.commit()
+        cur.close()
+        st.success("User added successfully!")
+    except Exception as e:
+        st.error(f"Error inserting user data: {e}")
+    finally:
+        conn.close()
 
+# Streamlit UI
 st.title("Add a New User")
 
 with st.form("user_form"):