Update app.py

app.py

@@ -35,11 +35,21 @@ async def optimize_text(prompt: InputPrompt):
     return {"optimized_text": optimized_text}
 
 def gen(prompt):
-    response = client.models.generate_content(
-        model="gemma-3-12b-it",
-        contents='<start_of_turn>system\n'+System_instruction+'<end_of_turn>\n<start_of_turn>user\n'+prompt+'<end_of_turn>\n<start_of_turn>model\n'
-    )
-    return response.text
+    try:
+        response = client.models.generate_content(
+            model="gemma-4-26b-a4b-it",
+            contents=[
+                types.Content(role="system", parts=[types.Part(text=System_instruction)]),
+                types.Content(role="user", parts=[types.Part(text=prompt)])
+            ],
+            config=types.GenerateContentConfig(
+                thinking_config=types.ThinkingConfig(thinking_level="MINIMAL")
+            ),
+        )
+        return response.text.rstrip()
+    except Exception as e:
+        print(f"GenAI Error: {e}")
+        raise HTTPException(status_code=500, detail="AI Generation Failed")
 
 @app.get("/", response_class=HTMLResponse)
 async def read_items():