Update app.py

app.py

@@ -2,8 +2,7 @@ import streamlit as st
 import os
 import json
 import requests
-from google import genai
-from google.genai import types
+import google.generativeai as genai  # Correct import
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -17,7 +16,7 @@ st.set_page_config(
 
 # Initialize Gemini client
 api_key = st.secrets["GOOGLE_API_KEY"]
-client = genai.Client(api_key=api_key)
+genai.configure(api_key=api_key)  # Correct initialization
 
 # Indian Stock Market API base configuration
 INDIAN_API_KEY = st.secrets["FINANCE_KEY"]
@@ -31,50 +30,7 @@ API_ENDPOINTS = {
         "param_mapping": {"stock_name": "name"},
         "description": "Get details for a specific stock"
     },
-    "get_trending_stocks": {
-        "endpoint": "/trending",
-        "required_params": [],
-        "param_mapping": {},
-        "description": "Get trending stocks in the market"
-    },
-    "get_market_news": {
-        "endpoint": "/news",
-        "required_params": [],
-        "param_mapping": {},
-        "description": "Get latest stock market news"
-    },
-    "get_mutual_funds": {
-        "endpoint": "/mutual_funds",
-        "required_params": [],
-        "param_mapping": {},
-        "description": "Get mutual funds data"
-    },
-    "get_ipo_data": {
-        "endpoint": "/ipo",
-        "required_params": [],
-        "param_mapping": {},
-        "description": "Get IPO data"
-    },
-    "get_bse_most_active": {
-        "endpoint": "/BSE_most_active",
-        "required_params": [],
-        "param_mapping": {},
-        "description": "Get BSE most active stocks"
-    },
-    "get_nse_most_active": {
-        "endpoint": "/NSE_most_active",
-        "required_params": [],
-        "param_mapping": {},
-        "description": "Get NSE most active stocks"
-    },
-    "get_historical_data": {
-        "endpoint": "/historical_data",
-        "required_params": ["stock_name"],
-        "optional_params": ["period"],
-        "default_values": {"period": "1m", "filter": "default"},
-        "param_mapping": {},
-        "description": "Get historical data for a stock"
-    }
+    # ... (rest of your code)
 }
 
 # Unified API call function
@@ -183,28 +139,16 @@ def orchestrator(query):
         
     # Create content for the orchestrator
     contents = [
-        types.Content(
-            role="user",
-            parts=[
-                types.Part.from_text(text=orchestrator_prompt)
-            ],
-        ),
+        {
+            "role": "user",
+            "parts": [orchestrator_prompt]
+        }
     ]
     
-    # Configure generation parameters
-    generate_content_config = types.GenerateContentConfig(
-        temperature=0.2,
-        top_p=0.95,
-        top_k=40,
-        max_output_tokens=500,
-        response_mime_type="text/plain",
-    )
-    
     # Generate content
-    response = client.models.generate_content(
+    response = genai.generate_content(
         model="gemini-1.5-flash",
         contents=contents,
-        config=generate_content_config,
     )
     
     # Parse the response
@@ -223,7 +167,6 @@ def orchestrator(query):
     except Exception as e:
         print(f"Error parsing orchestrator response: {e}")
         return {"needs_api": False}
-
 # Language setting
 
 # Financial assistant system prompt