Debug Added

utils.py

@@ -176,10 +176,10 @@ class SafetyGuard:
     """Validates input and filters output"""
 
     def __init__(self):
-        self.financial_terms = {
-            'revenue', 'profit', 'ebitda', 'balance', 'cash',
-            'income', 'fiscal', 'growth', 'margin', 'expense'
-        }
+        # self.financial_terms = {
+        #     'revenue', 'profit', 'ebitda', 'balance', 'cash',
+        #     'income', 'fiscal', 'growth', 'margin', 'expense'
+        # }
         self.blocked_topics = {
             'politics', 'sports', 'entertainment', 'religion',
             'medical', 'hypothetical', 'opinion', 'personal'
@@ -187,10 +187,10 @@ class SafetyGuard:
 
     def validate_input(self, query: str) -> Tuple[bool, str]:
         query_lower = query.lower()
+        # if not any(term in query_lower for term in self.financial_terms):
+        #     return False, "Please ask financial questions."
         if any(topic in query_lower for topic in self.blocked_topics):
             return False, "I only discuss financial topics."
-        if not any(term in query_lower for term in self.financial_terms):
-            return False, "Please ask financial questions."
         return True, ""
 
     def filter_output(self, response: str) -> str:
@@ -266,18 +266,26 @@ Context: {context}<|im_end|>
 Answer:"""
 
         print(f"\n\n[For Debug Only] Prompt: {prompt}\n\n")
-
         response = generator(prompt)[0]['generated_text']
+        print(f"\n\n[For Debug Only] response: {response}\n\n")
+        
         clean_response = extract_final_response(response)
         clean_response = guard.filter_output(clean_response)
-
+        print(f"\n\n[For Debug Only] clean_response: {clean_response}\n\n")
+        
         query_embed = embeddings.embed_query(query)
+        print(f"\n\n[For Debug Only] query_embed: {query_embed}\n\n")
+        
         response_embed = embeddings.embed_query(clean_response)
+        print(f"\n\n[For Debug Only] response_embed: {response_embed}\n\n")
+        
         confidence = cosine_similarity([query_embed], [response_embed])[0][0]
-
-        memory.add_interaction(query, clean_response)
-
+        print(f"\n\n[For Debug Only] confidence: {confidence}\n\n")
+        
+        memory.add_interaction(query, clean_response)       
+        
+        print(f"\n\n[For Debug Only] I'm Done \n\n")
         return clean_response, round(confidence, 2)
 
     except Exception as e:
-        return f"Error processing request: {e}", 0.0
+        return f"Error processing request: {e}", 0.0