Update app.py

app.py

@@ -170,24 +170,16 @@ def orchestrator(query):
         }
     }
     """
-        
-    # Create content for the orchestrator
-    contents = [
-        {
-            "role": "user",
-            "parts": [orchestrator_prompt]
-        }
-    ]
     
     # Generate content
-    response = genai.generate_content(
+    response = genai.generate_text(
         model="gemini-1.5-flash",
-        contents=contents,
+        prompt=orchestrator_prompt,
     )
     
     # Parse the response
     try:
-        decision_text = response.text
+        decision_text = response.result
         # Extract JSON from the response (it might be wrapped in markdown code blocks)
         if "```json" in decision_text:
             json_str = decision_text.split("```json")[1].split("```")[0].strip()
@@ -319,8 +311,8 @@ if prompt := st.chat_input("Ask me anything about finance or investing..."):
             # Stream the response
             response_stream = client.models.generate_content_stream(
                 model="gemini-1.5-flash",
-                contents=contents,
-                config=generate_content_config,
+                prompt=user_query,
+                stream=True,
             )
             
             # Process streaming response