Update app.py

app.py

@@ -38,6 +38,9 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 # Database configuration
 DATABASE_PATH = '/tmp/chat_database.db'
 
+llm = None
+llm_chain = None
+
 # Initialize prompt template
 prompt_template = """
 Role: You are Figr Code Assistant, specializing in providing clear, error-free Python code solutions.
@@ -265,6 +268,26 @@ def init_db():
         print(f"Error initializing database: {e}")
         raise
 
+def initialize_app():
+    """Initialize all components of the application"""
+    global llm, llm_chain
+    
+    try:
+        print("Initializing database...")
+        init_db()
+        print("Database initialized successfully")
+        
+        print("Starting model initialization...")
+        llm = initialize_model()
+        print("Model initialized successfully")
+        
+        print("Creating LLM chain...")
+        llm_chain = LLMChain(llm=llm, prompt=prompt)
+        print("LLM chain created successfully")
+    except Exception as e:
+        print(f"Initialization error: {e}")
+        raise
+
         
 # Rest of the prompt template and other configurations remain the same
 prompt_template = """
@@ -445,6 +468,12 @@ def get_chat_list():
 
 @app.route("/api/chat", methods=["POST"])
 def chat():
+    global llm_chain
+    if llm_chain is None:
+        return jsonify({
+            "response": "System is still initializing. Please try again in a moment.",
+            "success": False
+        })
     data = request.json
     user_input = data.get("message", "")
     session_id = data.get("sessionId", "default")
@@ -691,21 +720,7 @@ def home():
     return render_template("index.html")
 
 
-try:
-    print("Initializing database...")
-    init_db()
-    print("Database initialized successfully")
-    
-    print("Starting model initialization...")
-    llm = initialize_model()
-    print("Model initialized successfully")
-    
-    print("Creating LLM chain...")
-    llm_chain = LLMChain(llm=llm, prompt=prompt)
-    print("LLM chain created successfully")
-except Exception as e:
-    print(f"Initialization error: {e}")
-    raise
+initialize_app()
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=PORT)