Update app.py

app.py

@@ -52,23 +52,6 @@ class ChatHistory:
     def clear(self):
         self.messages = []
 
-# Function to log questions and answers to a file
-def log_to_file(question: str, answer: str):
-    """Logs the question and answer to a file with a timestamp."""
-    try:
-        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        log_entry = f"Timestamp: {timestamp}\nQuestion: {question}\nAnswer: {answer}\n\n"
-        
-        # Debugging: Print the log entry to verify its content
-        logger.info(f"Log Entry: {log_entry}")
-        
-        # Open the file in append mode and write the log entry
-        with open("Logs.txt", "a") as log_file:
-            log_file.write(log_entry)
-            logger.info("Successfully wrote to Logs.txt")
-    except Exception as e:
-        logger.error(f"Failed to write to Logs.txt: {e}")
-
 # Load environment variables and setup
 load_dotenv()
 
@@ -92,6 +75,7 @@ except Exception as e:
     logger.error("Failed to connect to Qdrant.")
     exit(1)
 
+# Create the main collection for Mawared HR
 collection_name = "mawared"
 
 try:
@@ -118,9 +102,54 @@ retriever = db.as_retriever(
     search_kwargs={"k": 5}
 )
 
+# Create a new collection for logs
+logs_collection_name = "mawared_logs"
+
+try:
+    client.create_collection(
+        collection_name=logs_collection_name,
+        vectors_config=models.VectorParams(
+            size=384,  # Same size as embeddings
+            distance=models.Distance.COSINE
+        )
+    )
+    logger.info(f"Created new Qdrant collection: {logs_collection_name}")
+except Exception as e:
+    if "already exists" not in str(e):
+        logger.error(f"Error creating logs collection: {e}")
+        exit(1)
+
+def log_to_qdrant(question: str, answer: str):
+    """Logs the question and answer to the Qdrant logs collection."""
+    try:
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        log_entry = {
+            "question": question,
+            "answer": answer,
+            "timestamp": timestamp
+        }
+        
+        # Convert the log entry to a vector (using embeddings)
+        log_vector = embeddings.embed_documents([str(log_entry)])[0]
+        
+        # Insert the log into the Qdrant collection
+        client.upsert(
+            collection_name=logs_collection_name,
+            points=[
+                models.PointStruct(
+                    id=hash(timestamp),  # Use timestamp hash as ID
+                    vector=log_vector,
+                    payload=log_entry
+                )
+            ]
+        )
+        logger.info(f"Logged question and answer to Qdrant collection: {logs_collection_name}")
+    except Exception as e:
+        logger.error(f"Failed to log to Qdrant: {e}")
+
 llm = ChatGoogleGenerativeAI(
     model="gemini-2.0-flash-thinking-exp-01-21",
-    temperature=0.2,
+    temperature=0,
     max_tokens=None,
     timeout=None,
     max_retries=2,
@@ -131,7 +160,6 @@ llm = ChatGoogleGenerativeAI(
 template = """
 You are a specialized AI assistant for the Mawared HR System, designed to deliver accurate and contextually relevant support based solely on the provided context and chat history.
 
-
 ---
 
 Core Principles
@@ -196,8 +224,8 @@ Critical Constraints
 
 - Strict Context Reliance: Base all responses solely on the provided context and chat history.
 - Non-Mawared HR Queries: Politely decline to answer questions unrelated to Mawared HR.
-- Answer Format: Always provide accurate answers in numbered steps without using code.
-- Use Tags like <Thinking> and <Processing> to indicate your process.
+- Answer Format: Always provide accurate answers in numbered steps without revealing your thought process or using code.
+
 ---
 
 By adhering to these principles and guidelines, ensure every response is accurate, professional, and easy to follow.
@@ -277,9 +305,9 @@ def ask_question_gradio(question: str, history: List[List[str]]) -> Generator[tu
         # Add final response to chat history
         chat_history.add_message("assistant", response)
         
-        # Log the question and answer to a file
-        logger.info("Attempting to log question and answer to Logs.txt")
-        log_to_file(question, response)
+        # Log the question and answer to Qdrant
+        logger.info("Attempting to log question and answer to Qdrant")
+        log_to_qdrant(question, response)
         
     except Exception as e:
         logger.error(f"Error during question processing: {e}")