Update app.py

app.py

@@ -11,7 +11,7 @@ import tempfile
 
 # Initialize Google Gemini AI client
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
-client = genai.Client()
+client = genai.GenerativeModel('gemini-2.0-flash-thinking-exp')
 
 def convert_excel_to_csv(excel_file):
     """Convert Excel file to CSV and return the DataFrame"""
@@ -46,8 +46,7 @@ def analyze_columns(df: pd.DataFrame, filename: str) -> dict:
     """
     
     try:
-        response = client.models.generate_content(
-            model="gemini-2.0-flash-thinking-exp",
+        response = client.generate_content(
             contents=[
                 types.Part.from_text(f"Filename: {filename}\n\nCSV Data:\n{sample_csv}"),
                 analysis_prompt