Added updated intent

main.py

@@ -1,6 +1,8 @@
 # main.py
 import asyncio
 import importlib
+from datetime import datetime
+import requests
 from voice.speech_to_text import SpeechToText
 from voice.intent_classifier import IntentClassifier
 # from voice.classifier import TextClassifier
@@ -111,7 +113,7 @@ async def process_query(vosk_model_path, audio_data=None, query_text=None, use_r
                             output["retriever_response"] = retriever_response
                     else:
                         # Base response failed, use the retriever
-                        output["base_response"] = f"{base_response}. Using retriever to query CSV file..."
+                        output["base_response"] = f"{base_response} Using retriever to query CSV file..."
                         # retriever_response = retriever.retrieve(text, entities)
                         # retriever_response = sql_db.keyword_match_search(entities)
                         retriever_response = sql_db.query_db(entities["ticker"], entities["metric"])
@@ -144,6 +146,19 @@ async def process_query(vosk_model_path, audio_data=None, query_text=None, use_r
     except Exception as e:
         output["error"] = f"Unexpected error: {e}"
 
-    # print(output)
+
+    # Current Time
+    now = datetime.now()
+    print("Current Time:", now.strftime("%Y-%m-%d %H:%M:%S"))
+
+    # Location Info
+    try:
+        response = requests.get("https://ipinfo.io")
+        data = response.json()
+        print("Location:", data.get("city"), data.get("region"), data.get("country"))
+
+    except Exception as e:
+        print("Could not fetch location:", e)
+
     # Return output to the User Interface
     return output

rag/sql_db.py

@@ -166,7 +166,7 @@ class SQL_Key_Pair:
             if result:
                 value = result[0]
                 value_in_billions = value / 1_000_000_000
-                return f"{metric} for {ticker}: ${value_in_billions:.2f} billion."
+                return f"The {metric} for {ticker} was : ${value_in_billions:.2f} billion in 2024."
             return f"No relevant data found for {ticker}."
         except Exception as e:
             print(f"Error querying database: {e}")
@@ -176,6 +176,6 @@ class SQL_Key_Pair:
         try:
             if hasattr(self, 'db_conn') and self.db_conn:
                 self.db_conn.close()
-                print("Closed SQLite database connection")
+                # print("Closed SQLite database connection")
         except Exception as e:
             print(f"Error closing database connection: {e}")