Update app.py

app.py

@@ -25,7 +25,9 @@ from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 
-import google.generativeai as genai
+# Updated: Use the new Google Genai library
+from google import genai
+from google.genai import types
 from huggingface_hub import hf_hub_download
 
 # ---------------------------------------------------------------
@@ -163,15 +165,34 @@ def load_system_instruction(style="standard") -> str:
         return open(f, encoding="utf-8").read()
 
 def generate_html_from_text(text: str, temperature=0.5, style="standard") -> str:
+    # Updated: Use the new Google Genai client API
     api_key = os.environ["GEMINI_API_KEY"]
-    genai.configure(api_key=api_key)
-    model = genai.GenerativeModel(os.getenv("GEMINI_MODEL", "gemini-1.5-pro"))
+    client = genai.Client(api_key=api_key)
+    
+    model_name = os.getenv("GEMINI_MODEL", "gemini-1.5-pro")
     prompt = f"{load_system_instruction(style)}\n\n{text}"
-    cfg = dict(temperature=temperature, top_p=0.7, top_k=20, max_output_tokens=8192, candidate_count=1)
-    resp = model.generate_content(prompt, generation_config=cfg)
-    raw = resp.text
+    
+    # Configure generation parameters
+    config = types.GenerateContentConfig(
+        temperature=temperature,
+        top_p=0.7,
+        top_k=20,
+        max_output_tokens=8192,
+        candidate_count=1
+    )
+    
+    # Generate content
+    response = client.models.generate_content(
+        model=model_name,
+        contents=prompt,
+        config=config
+    )
+    
+    # Extract HTML from response
+    raw = response.text
     s, e = raw.find("```html"), raw.rfind("```")
     html = raw[s+7:e].strip() if s != -1 and e != -1 else raw
+    
     return enhance_font_awesome_layout(html)
 
 def trim_image_whitespace(img: Image.Image, threshold=248, padding=20) -> Image.Image: