Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, session
 from huggingface_hub import InferenceClient
 import os
 import random
@@ -8,9 +8,11 @@ keys = [
     os.getenv("KEY2"), 
     os.getenv("KEY3"), 
     os.getenv("KEY4"), 
-    os.getenv("KEY5")]
+    os.getenv("KEY5")
+]
 
 app = Flask(__name__)
+app.secret_key = os.urandom(24)  # Set a secret key for session management
 
 client = InferenceClient(
     "meta-llama/Meta-Llama-3-8B-Instruct",
@@ -25,8 +27,15 @@ def home():
 def chat():
     user_message = request.json['message']
     
-    # Initial prompt structure
-    prompt = f"""<file>Frequently Asked Questions (FAQs)
+    # Initialize or retrieve the conversation history
+    if 'history' not in session:
+        session['history'] = []
+    
+    # Add the user's message to the history
+    session['history'].append(f"<user>{user_message}</user>")
+    
+    # Construct the prompt with the entire conversation history
+    prompt = """<file>Frequently Asked Questions (FAQs)
 
 What are the primary responsibilities of RAC?
 
@@ -178,12 +187,21 @@ You can apply for summer training to the Director of a suitable
 
 DRDO laboratory or institution. The procedure typically involves submitting an application detailing your academic background, research interests, and proposed project.
 </file>
-<system> Act as a DRDO FAQ chatbot name Tannu. Reply the user's queries based on the FAQ document.</system>
-<user>{user_message}</user>
-<Tannu>"""
+system>Act as a DRDO FAQ chatbot named Tannu. Reply to the user's queries based on the FAQ document.</system>\n"""
+    prompt += "\n".join(session['history'])
+    prompt += "\n<Tannu>"
 
     response = client.text_generation(prompt, max_new_tokens=1024, stop_sequences=["<user>"])
     
+    # Add the bot's response to the history
+    session['history'].append(f"<Tannu>{response}</Tannu>")
+    
+    # Trim history if it gets too long (optional, adjust the number as needed)
+    if len(session['history']) > 10:
+        session['history'] = session['history'][-10:]
+    
+    session.modified = True  # Ensure the session is saved
+    
     return jsonify({'message': response})
 
 if __name__ == '__main__':