Update app.py

app.py

@@ -8,18 +8,18 @@ openai.api_key = os.getenv("API_KEY")
 openai.api_base = "https://api.groq.com/openai/v1"
 
 # Import datasets from the Python files in your project
-from datasets.company_profile import company_profile
-from datasets.workforce import workforce
-from datasets.financials import financials
-from datasets.investors import investors
-from datasets.products_services import products_services
-from datasets.market_trends import market_trends
-from datasets.partnerships_collaborations import partnerships_collaborations
-from datasets.legal_compliance import legal_compliance
-from datasets.customer_insights import customer_insights
-from datasets.news_updates import news_updates
-from datasets.social_media import social_media
-from datasets.tech_stack import tech_stack
+from company_profile import company_profile
+from workforce import workforce
+from financials import financials
+from investors import investors
+from products_services import products_services
+from market_trends import market_trends
+from partnerships_collaborations import partnerships_collaborations
+from legal_compliance import legal_compliance
+from customer_insights import customer_insights
+from news_updates import news_updates
+from social_media import social_media
+from tech_stack import tech_stack
 
 # Command handler for specific queries
 def command_handler(user_input):
@@ -88,6 +88,14 @@ def market_analysis_agent(user_input, history=[]):
             response = investors
         elif "products" in user_input.lower():
             response = products_services
+        elif "news" or "updates" in user_input.lower():
+            response = news_updates
+        elif "legal" or "compliance" in user_input.lower():
+            response = legal_compliance
+        elif "social media" or "instagram" or "linkedin" or "twitter" in user_input.lower():
+            response = social_media
+        elif "product" or "services" in user_input.lower():
+            response = products_services
         elif "workforce" in user_input.lower():
             response = workforce
         else: