Update app.py

app.py

@@ -10,8 +10,8 @@ from huggingface_hub import (
 )
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
 from google import genai
-# Import Content specifically if we were using Option B, but sticking to Option A
-# from google.genai.types import Content
+# Import Content type for structured multi-turn/multi-role inputs
+from google.genai.types import Content
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
 
 
@@ -189,23 +189,29 @@ def run_agent(client, model_name, system_prompt_template, user_input_state, conf
     # Prepare the message content by formatting the project state
     user_message_content = "Project State:\n" + json.dumps(user_input_state, indent=2)
 
-    # Combine system prompt and user state into a single string and pass as a list of strings
-    prompt = system_prompt + "\n\n" + user_message_content
+    # Define the model to use based on the parameter
+    model_to_use = model_name
 
     try:
-        # Define model_to_use by using the passed model_name parameter
-        model_to_use = model_name
+        # *** FIX: Use Content objects for system and user roles ***
+        messages = [
+            Content(role="system", parts=[system_prompt]), # System instruction as a part
+            Content(role="user", parts=[user_message_content]) # User state as a part
+        ]
 
-        # Pass a list containing the single combined prompt string
         response = client.models.generate_content(
             model=model_to_use,
-            contents=[prompt], # Pass a list of strings
+            contents=messages, # Pass the list of Content objects
             config=config
         )
-        # *** FIX: REMOVED response.raise_for_status() as it doesn't exist on GenerateContentResponse ***
         # API errors are handled by the SDK raising exceptions caught below.
 
         # Some models return parts, concatenate them
+        # Ensure candidate and content exist before accessing parts
+        if not response.candidates or not response.candidates[0].content:
+             print("Agent returned no candidates or empty content.")
+             return f"ERROR: Agent returned no response content."
+
         response_text = "".join([part.text for part in response.candidates[0].content.parts])
 
         print(f"--- Agent Response --- ({model_to_use})")
@@ -246,7 +252,7 @@ def run_planner(client, project_state, config):
     }
     response_text = run_agent(
         client=client,
-        model_name="gemini-1.5-flash-latest", # Pass the model name
+        model_name="gemini-2.5-flash-preview-04-17", # *** FIX: Use the specific model name ***
         system_prompt_template=SYSTEM_ARCHITECT,
         user_input_state=input_state_for_planner,
         config=config,
@@ -277,7 +283,7 @@ def run_codegen(client, project_state, config):
     }
     response_text = run_agent(
         client=client,
-        model_name="gemini-1.5-flash-latest", # Pass the model name
+        model_name="gemini-2.5-flash-preview-04-17", # *** FIX: Use the specific model name ***
         system_prompt_template=SYSTEM_CODEGEN,
         user_input_state=input_state_for_codegen,
         config=config,
@@ -369,7 +375,7 @@ def run_debugger(client, project_state, config):
     }
     response_text = run_agent(
         client=client,
-        model_name="gemini-1.5-flash-latest", # Pass the model name
+        model_name="gemini-2.5-flash-preview-04-17", # *** FIX: Use the specific model name ***
         system_prompt_template=SYSTEM_DEBUG,
         user_input_state=input_state_for_debugger,
         config=config,
@@ -859,7 +865,6 @@ with gr.Blocks(title="HF Space Auto‑Builder (Team AI)") as demo:
         outputs=[chatbot, build_box, run_box, preview, project_status_md]
     )
 
-    # Update the submit handler - inputs match handle_user_message signature
     user_in.submit(
         fn=handle_user_message,
         inputs=[