Added updated intent

Dockerfile

@@ -1,4 +1,3 @@
-# Use official Python image
 FROM python:3.10-slim
 
 # Install system dependencies
@@ -6,14 +5,12 @@ RUN apt-get update && \
     apt-get install -y wget ffmpeg unzip curl gcc  && \
     rm -rf /var/lib/apt/lists/*
 
-   # Set working directory
-   WORKDIR /app
+WORKDIR /app
 
-   # Set Hugging Face cache directory
-   ENV HF_HOME=/app/cache
+ENV HF_HOME=/app/cache
 
-   # Create cache directory and set permissions
-   RUN mkdir -p /app/cache /app/db && chmod -R 777 /app/cache /app/db
+# Create cache directory and set permissions
+RUN mkdir -p /app/cache /app/db && chmod -R 777 /app/cache /app/db
 
 # Copy your code into the container
 COPY . .
@@ -22,9 +19,6 @@ COPY . .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 RUN pip install --no-cache-dir torch==2.2.0+cpu torchvision==0.17.0+cpu torchaudio==2.2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
-
-
-
 # Install spaCy model
 RUN python -m spacy download en_core_web_lg
 # RUN python -m spacy download en_core_web_sm

main.py

@@ -55,22 +55,22 @@ async def process_query(vosk_model_path, audio_data=None, query_text=None, use_r
 
         if intent:
             intent_to_module = {
-                "get_net_income": ("modules.get_net_income", "GetNetIncome"),
-                "get_revenue": ("modules.get_revenue", "GetRevenue"),
-                "get_stock_price": ("modules.get_stock_price", "GetStockPrice"),
-                "get_profit_margin": ("modules.get_profit_margin", "GetProfitMargin"),
-                "get_company_profile": ("modules.get_company_profile", "GetCompanyProfile"),
-                "get_market_cap": ("modules.get_market_cap", "GetMarketCap"),
-                "get_historical_stock_price": ("modules.get_historical_stock_price", "GetHistoricalStockPrice"),
-                "get_dividend_info": ("modules.get_dividend_info", "GetDividendInfo"),
-                "get_balance_sheet": ("modules.get_balance_sheet", "GetBalanceSheet"),
-                "get_cash_flow": ("modules.get_cash_flow", "GetCashFlow"),
-                "get_financial_ratios": ("modules.get_financial_ratios", "GetFinancialRatios"),
-                "get_earnings_per_share": ("modules.get_earnings_per_share", "GetEarningsPerShare"),
-                "get_interest": ("modules.get_interest", "GetInterest"),
-                "get_income_tax": ("modules.get_income_tax", "GetIncomeTax"),
-                "get_cost_info": ("modules.get_cost_info", "GetCostInfo"),
-                "get_research_info": ("modules.get_research_info", "GetResearchInfo")
+                "net_income": ("modules.get_net_income", "GetNetIncome"),
+                "revenue": ("modules.get_revenue", "GetRevenue"),
+                "stock_price": ("modules.get_stock_price", "GetStockPrice"),
+                "profit_margin": ("modules.get_profit_margin", "GetProfitMargin"),
+                "company_info": ("modules.get_company_profile", "GetCompanyProfile"),
+                "market_capitalization": ("modules.get_market_cap", "GetMarketCap"),
+                "historical_stock_price": ("modules.get_historical_stock_price", "GetHistoricalStockPrice"),
+                "dividend_info": ("modules.get_dividend_info", "GetDividendInfo"),
+                "balance_sheet": ("modules.get_balance_sheet", "GetBalanceSheet"),
+                "cash_flow": ("modules.get_cash_flow", "GetCashFlow"),
+                "financial_ratios": ("modules.get_financial_ratios", "GetFinancialRatios"),
+                "earnings_per_share": ("modules.get_earnings_per_share", "GetEarningsPerShare"),
+                "interest_rate": ("modules.get_interest", "GetInterest"),
+                "income_tax": ("modules.get_income_tax", "GetIncomeTax"),
+                "cost_info": ("modules.get_cost_info", "GetCostInfo"),
+                "research_info": ("modules.get_research_info", "GetResearchInfo")
                 
             }
 

voice/intent_classifier.py

@@ -12,22 +12,22 @@ class IntentClassifier:
         self.nlp = spacy.load("en_core_web_lg")  # "en_core_web_sm"
         self.classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli", from_pt=True)
         self.intents = [
-            "get_net_income",
-            "get_revenue",
-            "get_stock_price",
-            "get_profit_margin",
-            "get_company_profile",
-            "get_market_cap",
-            "get_historical_stock_price",
-            "get_dividend_info",
-            "get_balance_sheet",
-            "get_cash_flow",
-            "get_financial_ratios",
-            "get_earnings_per_share",
-            "get_interest",
-            "get_research_info",
-            "get_cost_info",
-            "get_income_tax"
+            "net_income",
+            "revenue",
+            "stock_price",
+            "profit_margin",
+            "company_info",
+            "market_capitalization",
+            "historical_stock_price",
+            "dividend_info",
+            "balance_sheet",
+            "cash_flow",
+            "financial_ratios",
+            "earnings_per_share",
+            "interest_rate",
+            "research_info",
+            "cost_info",
+            "income_tax"
         ]
 
         # Mapping of company names to ticker symbols (case-insensitive)
@@ -168,7 +168,7 @@ class IntentClassifier:
 
         # Step 3: Extract metric using keyword matching with synonyms
         text_lower = text.lower()
-        if any(keyword in text_lower for keyword in ["net income", "net", "income"]):
+        if any(keyword in text_lower for keyword in ["income"]):
             entities["metric"] = "netIncome"
         elif "revenue" in text_lower:
             entities["metric"] = "revenue"
@@ -176,19 +176,19 @@ class IntentClassifier:
             entities["metric"] = "netProfitMargin"
         elif any(keyword in text_lower for keyword in ["market cap", "market capitalization", "market"]):
             entities["metric"] = "mktCap"
-        elif any(keyword in text_lower for keyword in ["payout ratio", "dividend payout"]):
+        elif any(keyword in text_lower for keyword in ["payout ratio", "payout", "dividend" "dividend payout"]):
             entities["metric"] = "payoutRatio"
         elif any(keyword in text_lower for keyword in ["current ratio", "liquidity ratio"]):
             entities["metric"] = "currentRatio"
-        elif any(keyword in text_lower for keyword in ["eps", "earnings per share", "earnings"]):
+        elif any(keyword in text_lower for keyword in ["eps", "earning", "earnings per share", "earnings"]):
             entities["metric"] = "eps"
         elif any(keyword in text_lower for keyword in ["stock", "stock price", "current price", "valuation", "price"]):
             entities["metric"] = "price"
-        elif any(keyword in text_lower for keyword in ["company info", "about company", "who is"]):
+        elif any(keyword in text_lower for keyword in ["company info", "company", "about company", "who is"]):
             entities["metric"] = "ceo"
-        elif any(keyword in text_lower for keyword in ["balance sheet", "sheet", "assets"]):
+        elif any(keyword in text_lower for keyword in ["balance sheet", "balance", "sheet", "assets"]):
             entities["metric"] = "Assets&Liabilities"
-        elif any(keyword in text_lower for keyword in ["historical", "earnings per share", "earnings"]):
+        elif any(keyword in text_lower for keyword in ["historical"]):
             entities["metric"] = "historical"
         elif any(keyword in text_lower for keyword in ["cash", "flow", "cash flow"]):
             entities["metric"] = "cashFlowFromOperatingActivities"