update signin

pytrade.py

@@ -17,7 +17,10 @@ import time
 import requests
 from typing import List, Dict
 from assistant import get_answer
-from signin import get_db_connection,generate_password_hash
+import pyodbc
+from werkzeug.security import generate_password_hash
+from signin import get_db_connection
+
 
 app = Flask(__name__)
 

signin.py

@@ -1,58 +1,33 @@
+# db.py
 import os
 import pyodbc
-from flask import Flask, request, jsonify
-from werkzeug.security import generate_password_hash, check_password_hash
-from flask_cors import CORS
 
-app = Flask(__name__)
-
-# -----------------------------------------------
-# CORS: allow multiple origins via env (optional)
-# -----------------------------------------------
-ALLOWED_ORIGINS = os.getenv("ALLOWED_ORIGINS", "*")
-CORS(app, resources={r"/*": {"origins": [o.strip() for o in ALLOWED_ORIGINS.split(",")]}})
-
-# ------------------------------------------------------
-# MODE: "local" uses Windows Auth; "server" uses SQL Auth
-# Set MODE=server in Hugging Face Variables & secrets
-# ------------------------------------------------------
 MODE = os.getenv("MODE", "local").lower()
 
-# ---------------------------
 # Local (Windows) connection
-# ---------------------------
 LOCAL_SQL_SERVER   = os.getenv("LOCAL_SQL_SERVER", r"localhost\SQLEXPRESS")
 LOCAL_SQL_DATABASE = os.getenv("LOCAL_SQL_DATABASE", "PyTrade")
 LOCAL_SQL_DRIVER   = os.getenv("LOCAL_SQL_DRIVER", "{ODBC Driver 17 for SQL Server}")
 
-# ----------------------------------------
-# Remote (HF/AWS RDS) SQL Auth connection
-# ----------------------------------------
-RDS_SQL_SERVER   = os.getenv("RDS_SQL_SERVER", "")          # e.g. mydb.abcxyz.ap-south-1.rds.amazonaws.com,1433
+# Remote (RDS/HF) SQL Auth
+RDS_SQL_SERVER   = os.getenv("RDS_SQL_SERVER", "")
 RDS_SQL_DATABASE = os.getenv("RDS_SQL_DATABASE", "PyTrade")
 RDS_SQL_USER     = os.getenv("RDS_SQL_USER", "")
 RDS_SQL_PASSWORD = os.getenv("RDS_SQL_PASSWORD", "")
 RDS_SQL_DRIVER   = os.getenv("RDS_SQL_DRIVER", "{ODBC Driver 18 for SQL Server}")
-RDS_ENCRYPT      = os.getenv("RDS_ENCRYPT", "yes")          # yes/no
-RDS_TRUST_CERT   = os.getenv("RDS_TRUST_SERVER_CERT", "yes")# yes/no
+RDS_ENCRYPT      = os.getenv("RDS_ENCRYPT", "yes")
+RDS_TRUST_CERT   = os.getenv("RDS_TRUST_SERVER_CERT", "yes")
 
-# ======================================================
-# Establishing the database connection using env values
-# (CORE BEHAVIOR UNCHANGED for queries)
-# ======================================================
 def get_db_connection():
     if MODE == "local":
-        # Windows Authentication (local)
-        connection = pyodbc.connect(
+        return pyodbc.connect(
             f"DRIVER={LOCAL_SQL_DRIVER};"
             f"SERVER={LOCAL_SQL_SERVER};"
             f"DATABASE={LOCAL_SQL_DATABASE};"
             f"Trusted_Connection=yes;"
         )
-        return connection
     else:
-        # SQL Authentication (RDS / Hugging Face)
-        connection = pyodbc.connect(
+        return pyodbc.connect(
             f"DRIVER={RDS_SQL_DRIVER};"
             f"SERVER={RDS_SQL_SERVER};"
             f"DATABASE={RDS_SQL_DATABASE};"
@@ -60,36 +35,23 @@ def get_db_connection():
             f"Encrypt={RDS_ENCRYPT};TrustServerCertificate={RDS_TRUST_CERT};"
             f"Connection Timeout=30;"
         )
-        return connection
 
-# ======================================================
-# Create the User table only on local
-# (CORE CREATE SQL KEPT THE SAME)
-# ======================================================
-def create_user_table():
+def ensure_user_table_exists():
     conn = get_db_connection()
-    cursor = conn.cursor()
-    cursor.execute('''
-        IF NOT EXISTS (SELECT * FROM sysobjects WHERE name='Users' AND xtype='U')
-        CREATE TABLE Users (
-            id INT IDENTITY(1,1) PRIMARY KEY,
-            name NVARCHAR(120) NOT NULL,
-            phone NVARCHAR(50) NOT NULL,
-            email NVARCHAR(120) UNIQUE NOT NULL,
-            password NVARCHAR(255) NOT NULL
-        )
-    ''')
-    conn.commit()
-    cursor.close()
-    conn.close()
-
-# Initialize the table on startup ONLY IF local
-if MODE == "local":
-    create_user_table()
-
-# ===========================
-# DO NOT CHANGE: API ROUTES
-# ===========================
-
-
-
+    try:
+        cur = conn.cursor()
+        cur.execute('''
+            IF NOT EXISTS (SELECT * FROM sysobjects WHERE name='Users' AND xtype='U')
+            CREATE TABLE Users (
+                id INT IDENTITY(1,1) PRIMARY KEY,
+                name NVARCHAR(120) NOT NULL,
+                phone NVARCHAR(50) NOT NULL,
+                email NVARCHAR(120) UNIQUE NOT NULL,
+                password NVARCHAR(255) NOT NULL
+            )
+        ''')
+        conn.commit()
+    finally:
+        try: cur.close()
+        except: pass
+        conn.close()