Update app.py

app.py

@@ -154,27 +154,31 @@ def init_llm():
     if vector_db is None:
         return jsonify({'error': 'Please upload PDFs first'}), 400
 
+    # Get parameters from the incoming request
     data = request.json
-    model_name = data.get('model', 'llama')  # default to llama
+    model_name = data.get('model', 'llama')  # Default to 'llama' if not provided
     temperature = data.get('temperature', 0.5)
     max_tokens = data.get('max_tokens', 4096)
     top_k = data.get('top_k', 3)
 
+    # Ensure the model name is valid
     if model_name not in LLM_MODELS:
         return jsonify({'error': 'Invalid model name'}), 400
 
     try:
+        # Initialize the LLM chain with the specified parameters and the vector_db
         qa_chain = initialize_llmchain(
-            LLM_MODELS[model_name],
-            temperature,
-            max_tokens,
-            top_k,
-            vector_db
+            llm_model=LLM_MODELS[model_name],
+            temperature=temperature,
+            max_tokens=max_tokens,
+            top_k=top_k,
+            vector_db=vector_db  # Pass vector_db to the function
         )
         return jsonify({'message': 'LLM initialized successfully'}), 200
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+
 @app.route('/chat', methods=['POST'])
 def chat():
     """Handle chat interactions"""