Update app.py

app.py

@@ -76,11 +76,20 @@ except Exception as e:
     raise
 
 # Configure system message
-system_message = """You are AQuaBot, an AI assistant aware of environmental impact.
-You help users with any topic while raising awareness about water consumption
-in AI. Did you know that training large language models like Llama 2 can consume
-substantial amounts of water due to the cooling requirements of data centers?
-Let's work together while being mindful of our environmental impact."""
+system_message = """You are a helpful AI assistant called AQuaBot. You provide direct, clear answers to questions while being aware of environmental impact. Keep your responses natural and concise. Respond directly to questions without using any special tags or markers."""
+
+def generate_response(user_input, chat_history):
+    try:
+        logger.info("Generating response for user input...")
+        global total_water_consumption
+
+        # Calculate water consumption for input
+        input_water_consumption = calculate_water_consumption(user_input, True)
+        total_water_consumption += input_water_consumption
+
+        # Create prompt following Llama 2 chat format
+        conversation = f""<s>[INST] {system_message}
+
 
 # Constants for water consumption calculation
 WATER_PER_TOKEN = {