Update agents/compare.py

agents/compare.py

@@ -1,37 +1,37 @@
-import pandas as pd
-import google.generativeai as genai
-import os
-
-#function to compare cards
-def compare_selected_cards(selected_names, card_lookup):
-    genai.configure(api_key=os.environ.get("api_key_4"))  
-    model4 = genai.GenerativeModel('gemini-1.5-flash-latest')
-    print(selected_names)
-    if not selected_names or len(selected_names) < 2:
-        return "<b style='color:red;'>Please select at least two cards to compare.</b>"
-
-    comparison_data = "\n\n".join([
-            f"{name}: {card_lookup.get(name)}" for name in selected_names
-        ])
-    prompt = (
-            f"Do not include extra symbols like (* or #), just generate a textual response maybe with numbers for points."
-            f"Compare the following credit cards based on their benefits. "
-            f"Keep the output short, structured, and very readable:\n\n"
-            f"{comparison_data}\n\n"
-            f"Use markdown format with clear section headers like 'Comparison' and 'Recommendation'. "
-            f"Present key differences as bullet points without using '*' symbols — use '-' instead. "
-            f"Make it crisp and avoid lengthy explanations. "
-            f"Conclude with a recommendation on which card suits which type of user."
-            f"DO NOT INCLUDE ANY SYMBOLS LIKE # OR *"
-    )
-    print("comparing")
-    try:
-        response = model4.generate_content(prompt)
-        return f"""
-            <div style='background-color: #f9fbe7; padding: 15px; border-radius: 10px; font-family: sans-serif;'>
-                <pre style='white-space: pre-wrap; font-size: 13px; color: #333;'>{response.text}</pre>
-            </div>
-            """
-    except Exception as e:
-        print("Comparison Error:", e)
+import pandas as pd
+import google.generativeai as genai
+import os
+
+#function to compare cards
+def compare_selected_cards(selected_names, card_lookup):
+    genai.configure(api_key=os.environ.get("api_key_4"))  
+    model4 = genai.GenerativeModel('gemini-2.0-flash')
+    print(selected_names)
+    if not selected_names or len(selected_names) < 2:
+        return "<b style='color:red;'>Please select at least two cards to compare.</b>"
+
+    comparison_data = "\n\n".join([
+            f"{name}: {card_lookup.get(name)}" for name in selected_names
+        ])
+    prompt = (
+            f"Do not include extra symbols like (* or #), just generate a textual response maybe with numbers for points."
+            f"Compare the following credit cards based on their benefits. "
+            f"Keep the output short, structured, and very readable:\n\n"
+            f"{comparison_data}\n\n"
+            f"Use markdown format with clear section headers like 'Comparison' and 'Recommendation'. "
+            f"Present key differences as bullet points without using '*' symbols — use '-' instead. "
+            f"Make it crisp and avoid lengthy explanations. "
+            f"Conclude with a recommendation on which card suits which type of user."
+            f"DO NOT INCLUDE ANY SYMBOLS LIKE # OR *"
+    )
+    print("comparing")
+    try:
+        response = model4.generate_content(prompt)
+        return f"""
+            <div style='background-color: #f9fbe7; padding: 15px; border-radius: 10px; font-family: sans-serif;'>
+                <pre style='white-space: pre-wrap; font-size: 13px; color: #333;'>{response.text}</pre>
+            </div>
+            """
+    except Exception as e:
+        print("Comparison Error:", e)
         return "<b style='color:red;'>Something went wrong while comparing. Please try again.</b>"