Update app.py

app.py

@@ -2,25 +2,21 @@ import streamlit as st
 import os
 import json
 import requests
-from google import genai
-from google.genai import types
-from dotenv import load_dotenv
+import google.generativeai as genai
 
-load_dotenv()
+# Load environment variables
+api_key = os.getenv("GOOGLE_API_KEY")
+finance_key = os.getenv("FINANCE_KEY")
 
-# Set page configuration
-st.set_page_config(
-    page_title="Flaix - Financial Assistant",
-    page_icon="💰",
-    layout="centered"
-)
+if not api_key or not finance_key:
+    st.error("API keys not found. Please check your configuration.")
+    st.stop()
 
 # Initialize Gemini client
-api_key = st.secrets["GOOGLE_API_KEY"]
-client = genai.Client(api_key=api_key)
+genai.configure(api_key=api_key)
 
 # Indian Stock Market API base configuration
-INDIAN_API_KEY = st.secrets["FINANCE_KEY"]
+INDIAN_API_KEY = finance_key
 INDIAN_API_BASE_URL = "https://stock.indianapi.in"
 
 # Define API endpoints and their parameters
@@ -177,28 +173,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
@@ -218,6 +202,7 @@ def orchestrator(query):
         print(f"Error parsing orchestrator response: {e}")
         return {"needs_api": False}
 
+
 # Language setting
 
 # Financial assistant system prompt