Update app.py

app.py

@@ -46,7 +46,8 @@ TEMP_PATH = None
 #TABLE_PATH = None
 
 #System prompt
-PROMPT_TEMPLATE = """
+
+'''PROMPT_TEMPLATE = """
 You are working with a retrieval-augmented generation (RAG) setup. Your task is to generate a response based on the context provided and the question asked. Consider only the following context strictly, and use it to answer the question. If the question cannot be answered using the context, respond with: "The information requested is not mentioned in the context."
 
 Context:
@@ -59,26 +60,26 @@ Question:
 
 Response:
 """
+'''
 
+PROMPT_TEMPLATE = """
+You are working as a retrieval-augmented generation (RAG) assistant specializing in providing precise and accurate responses. Generate a response based only on the provided context and question, following these concrete instructions:
 
-# PROMPT_TEMPLATE = """
-# You are working with a retrieval-augmented generation (RAG) setup. Your task is to generate a response based on the provided context, table data, and the question asked. Consider only the given inputs strictly and use them to answer the question. Do not include any external information.
-
-# If the table variable contains tabular data, analyze and extract all relevant details from it. Provide a structured response summarizing the table data if it is relevant to the question. If the table data is not relevant, base your answer only on the context.
-
-# Context:
-# {context}
+- **Adhere strictly to the context:** Use only the information in the context to answer the question. Do not add any external details or assumptions.
+- **Handle multiple chunks:** The context is divided into chunks, separated by "###". Query-related information may be present in any chunk.
+- **Focus on relevance:** Identify and prioritize chunks relevant to the question while ignoring unrelated chunks.
+- **Answer concisely and factually:** Provide clear, direct, and structured responses based on the retrieved information.
 
-# Table:
-# {table}
+Context:
+{context}
 
-# ---
+---
 
-# Question:
-# {question}
+Question:
+{question}
 
-# Response:
-# """
+Response:
+"""
 
 #HFT = os.getenv('HF_TOKEN')
 #client = InferenceClient(api_key=HFT)
@@ -163,6 +164,7 @@ def chat():
 
     return render_template('chat.html', history=session['history'])
 
+'''
 @app.route('/create-db', methods=['GET', 'POST'])
 def create_db():
     if request.method == 'POST':
@@ -207,6 +209,45 @@ def create_db():
         return redirect(url_for('list_dbs'))
 
     return render_template('create_db.html')
+'''
+@app.route('/create-db', methods=['GET', 'POST'])
+def create_db():
+    if request.method == 'POST':
+        db_name = request.form['db_name']
+
+        # Ensure the upload folder exists
+        os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+
+        # Define the base upload path
+        upload_base_path = os.path.join(app.config['UPLOAD_FOLDER'], secure_filename(db_name))
+        os.makedirs(upload_base_path, exist_ok=True)
+
+        # Check for uploaded folder or files
+        folder_files = request.files.getlist('folder')
+        single_files = request.files.getlist('file')
+
+        if folder_files and any(file.filename for file in folder_files):
+            # Process folder files
+            for file in folder_files:
+                file_path = os.path.join(upload_base_path, secure_filename(file.filename))
+                os.makedirs(os.path.dirname(file_path), exist_ok=True)
+                file.save(file_path)
+
+        elif single_files and any(file.filename for file in single_files):
+            # Process single files
+            for file in single_files:
+                file_path = os.path.join(upload_base_path, secure_filename(file.filename))
+                file.save(file_path)
+
+        else:
+            return "No files uploaded", 400
+
+        # Generate datastore
+        generate_data_store(upload_base_path, db_name)
+
+        return redirect(url_for('list_dbs'))
+
+    return render_template('create_db.html')
 
 @app.route('/list-dbs', methods=['GET'])
 def list_dbs():