Update app.py

app.py

@@ -1,35 +1,6 @@
 import os
 import re
-import google.generativeai as genai
-
-class BasicAgent:
-
-    def __init__(self):
-        api_key = os.getenv("GEMINI_API_KEY")
-        if not api_key:
-            raise ValueError("Missing GEMINI_API_KEY")
-
-        genai.configure(api_key=api_key)
-
-        # ✅ stable model
-        self.model = genai.GenerativeModel("gemini-1.5-flash")
-
-        print("✅ Gemini Agent Ready")
-
-    def clean(self, text: str) -> str:
-        if not text:
-            return "N/A"
-
-        text = text.strip()
-        text = text.replace("Final Answer:", "").strip()
-        text = re.sub(r"\n.*", "", text)  # first line only
-        text = text.strip()
-
-        return text[:100]
-
-    import os
-import re
-import google.generativeai as genai
+from google import genai
 
 class BasicAgent: