rnrahate007 commited on
Commit
5a865d3
·
verified ·
1 Parent(s): 16f69a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -9
app.py CHANGED
@@ -35,17 +35,14 @@ class GeminiAgent:
35
  """
36
  try:
37
  time.sleep(6)
 
38
  response = self.model.generate_content(prompt)
39
  answer = response.text.strip()
40
- # Ensure we never return an empty or placeholder answer
41
- if not answer or answer.lower() in ["na", "n/a", "i don't know", "unknown"]:
42
- # Fallback: try a more specific prompt to force an answer
43
- fallback_prompt = f"Answer this question directly with a short, factual answer: {question}\nFinal Answer:"
44
- fallback_response = self.model.generate_content(fallback_prompt)
45
- answer = fallback_response.text.strip()
46
- if not answer:
47
- answer = "The answer could not be determined."
48
- print(f"GeminiAgent returning answer (first 50 chars): {answer[:50]}...")
49
  return answer
50
  except Exception as e:
51
  print(f"Gemini API error: {e}")
 
35
  """
36
  try:
37
  time.sleep(6)
38
+
39
  response = self.model.generate_content(prompt)
40
  answer = response.text.strip()
41
+
42
+ # simple clean (NO second call)
43
+ if not answer:
44
+ answer = "0" # or any short default
45
+
 
 
 
 
46
  return answer
47
  except Exception as e:
48
  print(f"Gemini API error: {e}")