CB commited on
Commit
0b25ca5
·
verified ·
1 Parent(s): 29eb03e

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +3 -3
streamlit_app.py CHANGED
@@ -15,6 +15,7 @@ from dotenv import load_dotenv
15
 
16
  load_dotenv()
17
 
 
18
  try:
19
  from phi.agent import Agent
20
  from phi.model.google import Gemini
@@ -435,7 +436,6 @@ if generate_now and not st.session_state.get("busy"):
435
  # Try 2: model-based interface (GenerativeModel / model.generate_content)
436
  if response is None:
437
  try:
438
- # prefer GenerativeModel if present
439
  if hasattr(genai, "GenerativeModel"):
440
  model_obj = genai.GenerativeModel(model=model_used)
441
  if hasattr(model_obj, "generate_content"):
@@ -458,7 +458,7 @@ if generate_now and not st.session_state.get("busy"):
458
  if response is None:
459
  raise RuntimeError("No supported generate method found on google.generativeai; check SDK version.")
460
 
461
- # Defensive normalization of response -> outputs list (keeps your existing extraction logic)
462
  outputs = []
463
  if response is None:
464
  outputs = []
@@ -483,7 +483,7 @@ if generate_now and not st.session_state.get("busy"):
483
  if not isinstance(outputs, list):
484
  outputs = list(outputs) if outputs else []
485
 
486
- # extract text pieces safely (identical to your prior logic)
487
  text_pieces = []
488
  for item in outputs:
489
  if item is None:
 
15
 
16
  load_dotenv()
17
 
18
+ # Optional phi/GenAI imports — gracefully degrade if not present
19
  try:
20
  from phi.agent import Agent
21
  from phi.model.google import Gemini
 
436
  # Try 2: model-based interface (GenerativeModel / model.generate_content)
437
  if response is None:
438
  try:
 
439
  if hasattr(genai, "GenerativeModel"):
440
  model_obj = genai.GenerativeModel(model=model_used)
441
  if hasattr(model_obj, "generate_content"):
 
458
  if response is None:
459
  raise RuntimeError("No supported generate method found on google.generativeai; check SDK version.")
460
 
461
+ # Defensive normalization of response -> outputs list
462
  outputs = []
463
  if response is None:
464
  outputs = []
 
483
  if not isinstance(outputs, list):
484
  outputs = list(outputs) if outputs else []
485
 
486
+ # extract text pieces safely
487
  text_pieces = []
488
  for item in outputs:
489
  if item is None: