Added print logs

core/models.py

@@ -43,6 +43,7 @@ class MutualFund(BaseModel):
         try:
             return cls.execute_raw_sql_query(query)
         except Exception as e:
+            print("Error: ", repr(e))
             return []
 
 class Stock(BaseModel):

core/text2sql/handler.py

@@ -21,6 +21,7 @@ class QueryDataHandler:
         """
         # Use Text2SQL ML model to generate a PostgreSQL query
         sql_query = self.text2sql_model.generate_query(prompt)
+        print("SQL Query: ", sql_query)
         logger.info(f"SQL Query: {sql_query}")
         # Use Django models to fetch data
         return sql_query, self.mutual_fund.execute_query(sql_query)

core/views.py

@@ -23,8 +23,8 @@ def get_mf_data(request):
     Retrieves mutual fund data based on user query.
     """
     query = request.GET.get("query", "")
-    logger.info(f"Query: {query}")
+    print("Query: ", query)
     prompt = get_prompt(query)
-    logger.info(f"Prompt: {prompt}")
+    print("Prompt: ", prompt)
     query, data = QueryDataHandler().get_data_from_query(prompt)
     return JsonResponse({"status": "success", "query": query, "data": data}, status=200)