removed session "

flask_app.py

@@ -1,6 +1,5 @@
 from flask import Flask, g, render_template, request, jsonify, session
 from flask_cors import CORS  # Add this import
-import os
 
 from helpers.GROQ import ConversationGROQ
 from helpers.postgres import DatabaseConnection
@@ -10,15 +9,13 @@ import pandas as pd
 
 app = Flask(__name__)
 CORS(app, supports_credentials=True, origins="https://db-bot-amber.vercel.app/", allow_headers=["Content-Type", "Authorization", "Access-Control-Allow-Credentials"], expose_headers=["Set-Cookie"])
-app.secret_key = os.urandom(24)  # Set a secret key for session encryption
-app.config['SESSION_COOKIE_SAMESITE'] = 'None'
-app.config['SESSION_COOKIE_SECURE'] = False 
-app.config['SESSION_COOKIE_HTTPONLY'] = True
+
 
 
 prompt_manager = PromptManager()
 prompt_manager.load_prompt('base_schema', 'prompts/base_prompts.txt')
 base_prompt = prompt_manager.get_prompt('base_schema')
+
 def extract_sql_regex(input_string) -> str | None:
     # Pattern to match SQL query within double quotes after "sql":
     pattern = r'"sql":\s*"(.*?)"'
@@ -29,27 +26,31 @@ def extract_sql_regex(input_string) -> str | None:
     else:
         return None
 
-# Add a function to get or create the database connection
-def get_db():
-    if 'db' not in g:
-        return DatabaseConnection(
-            db_host=session['DB_HOST'],
-            db_port=session['DB_PORT'],
-            db_name=session['DB_NAME'],
-            db_user=session['DB_USER'],
-            db_password=session['DB_PASSWORD']
-        )
-    return None
-
-@app.route('/', methods=['POST'])
-def index():
+
+
+@app.route('/chat', methods=['POST', 'OPTIONS'])  # Add OPTIONS method
+def chat():
+    # if request.method == 'OPTIONS':
+    #     # Respond to preflight request
+    #     response = app.make_default_options_response()
+    #     response.headers['Access-Control-Allow-Headers'] = 'Content-Type'
+    #     response.headers['Access-Control-Allow-Methods'] = 'POST'
+    #     return response
+
     data = request.json
+    
+    # if 'DB_HOST' not in session:
+    #     return jsonify({"error": "Database connection not established", "format": "json"}), 400
     db_user = data.get('DB_USER', '')
     db_host = data.get('DB_HOST', '')
     db_port = data.get('DB_PORT', '')
     db_name = data.get('DB_NAME', '')
     db_password = data.get('DB_PASSWORD', '')
+    prompt = data.get('prompt', '')
     missing_fields = []
+
+    if not prompt:
+        missing_fields.append('prompt')
     if not db_user:
         missing_fields.append('DB_USER')
     if not db_host:
@@ -67,42 +68,7 @@ def index():
             "format": "json"
         }), 400
 
-    # Store database credentials in session
-    session['DB_HOST'] = db_host
-    session['DB_PORT'] = db_port
-    session['DB_NAME'] = db_name
-    session['DB_USER'] = db_user
-    session['DB_PASSWORD'] = db_password
-
-    # Test the connection
-    try:
-        db = get_db()
-        if db is None:
-            return jsonify({"error": "Database connection failed", "format": "json"}), 500
-        return jsonify({"message": "Database connection successful", "format": "json"}), 200
-    except Exception as e:
-        return jsonify({"error": f"Database connection failed: {str(e)}", "format": "json"}), 500
-
-@app.route('/chat', methods=['POST', 'OPTIONS'])  # Add OPTIONS method
-def chat():
-    if request.method == 'OPTIONS':
-        # Respond to preflight request
-        response = app.make_default_options_response()
-        response.headers['Access-Control-Allow-Headers'] = 'Content-Type'
-        response.headers['Access-Control-Allow-Methods'] = 'POST'
-        return response
-
-    data = request.json
-    
-    if 'DB_HOST' not in session:
-        return jsonify({"error": "Database connection not established", "format": "json"}), 400
-    
-    prompt = data.get('prompt', '')
-    if not prompt:
-        return jsonify({"error": "Prompt is required", "format": "json"}), 400
-    
-    db = get_db()
-    
+    db = DatabaseConnection(db_host=db_host, db_port=db_port, db_name=db_name, db_user=db_user, db_password=db_password)
     schema = db.execute_query('SELECT schema_name FROM information_schema.schemata;').fetchall()
     schema = [schema[0] for schema in schema]