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 type for structured multi-turn/multi-role inputs
-from google.genai.types import Content
+# Import Content and Part types for structured multi-turn/multi-role inputs
+from google.genai.types import Content, Part
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
 
 
@@ -193,10 +193,10 @@ def run_agent(client, model_name, system_prompt_template, user_input_state, conf
     model_to_use = model_name
 
     try:
-        # *** FIX: Use Content objects for system and user roles ***
+        # *** FIX: Use Content objects with Part objects inside the parts list ***
         messages = [
-            Content(role="system", parts=[system_prompt]), # System instruction as a part
-            Content(role="user", parts=[user_message_content]) # User state as a part
+            Content(role="system", parts=[Part(text=system_prompt)]), # System instruction as a Part
+            Content(role="user", parts=[Part(text=user_message_content)]) # User state as a Part
         ]
 
         response = client.models.generate_content(
@@ -210,8 +210,14 @@ def run_agent(client, model_name, system_prompt_template, user_input_state, conf
         # 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.")
+             # Check if there was a rejection reason
+             if response.prompt_feedback and response.prompt_feedback.block_reason:
+                  block_reason = response.prompt_feedback.block_reason
+                  print(f"Prompt was blocked. Reason: {block_reason}")
+                  return f"ERROR: Agent response blocked by safety filters. Reason: {block_reason.name}" # Return reason if available
              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})")
@@ -252,7 +258,7 @@ def run_planner(client, project_state, config):
     }
     response_text = run_agent(
         client=client,
-        model_name="gemini-2.5-flash-preview-04-17", # *** FIX: Use the specific model name ***
+        model_name="gemini-2.5-flash-preview-04-17", # Use the specific model name
         system_prompt_template=SYSTEM_ARCHITECT,
         user_input_state=input_state_for_planner,
         config=config,
@@ -283,7 +289,7 @@ def run_codegen(client, project_state, config):
     }
     response_text = run_agent(
         client=client,
-        model_name="gemini-2.5-flash-preview-04-17", # *** FIX: Use the specific model name ***
+        model_name="gemini-2.5-flash-preview-04-17", # Use the specific model name
         system_prompt_template=SYSTEM_CODEGEN,
         user_input_state=input_state_for_codegen,
         config=config,
@@ -375,7 +381,7 @@ def run_debugger(client, project_state, config):
     }
     response_text = run_agent(
         client=client,
-        model_name="gemini-2.5-flash-preview-04-17", # *** FIX: Use the specific model name ***
+        model_name="gemini-2.5-flash-preview-04-17", # Use the specific model name
         system_prompt_template=SYSTEM_DEBUG,
         user_input_state=input_state_for_debugger,
         config=config,
@@ -435,21 +441,26 @@ def orchestrate_development(client, project_state, config, oauth_token_token):
 
         elif current_task.startswith('CODING'):
             # Ensure minimum files exist before asking CodeGen to code
-            # This happens once at the start of the first coding task
-            if project_state['attempt_count'] == 0 and project_state['current_task'] == 'CODING - Initial Implementation':
-                # Add initial stub files if they don't exist
-                if project_state['main_app_file'] not in project_state['files']:
+            # This happens once at the start of the first coding task or if syntax errors occurred
+            if project_state['attempt_count'] == 0 or 'Syntax Errors Found' in project_state.get('feedback', ''):
+                # Add initial stub files if they don't exist or need regeneration due to syntax issues
+                if project_state['main_app_file'] not in project_state['files'] or project_state.get('needs_initial_stubs', False): # Check flag if set
+                    print("Adding initial stubs for main app file...")
                     project_state['files'][project_state['main_app_file']] = f"# Initial {project_state['sdk_choice']} app file\n" # Start with a basic stub
                     if project_state['sdk_choice'] == 'gradio':
                          project_state['files'][project_state['main_app_file']] += "import gradio as gr\n\n# Define a simple interface\n# For example: gr.Interface(...).launch()\n"
                     elif project_state['sdk_choice'] == 'streamlit':
                            project_state['files'][project_state['main_app_file']] += "import streamlit as st\n\n# Your Streamlit app starts here\n# For example: st.write('Hello, world!')\n"
+                    project_state['needs_initial_stubs'] = False # Reset flag
+
 
                 if 'requirements.txt' not in project_state['files']:
+                     print("Adding initial requirements.txt stub...")
                      req_content = "pandas\n" + ("streamlit\n" if project_state['sdk_choice']=="streamlit" else "gradio\n") + "google-generativeai\nhuggingface-hub\n"
                      project_state['files']['requirements.txt'] = req_content
 
                 if 'README.md' not in project_state['files']:
+                    print("Adding initial README.md stub...")
                     readme_content = f"""---
 title: {project_state['repo_id']}
 emoji: 🐢
@@ -519,7 +530,7 @@ This is an auto-generated HF Space.
             # Wait a moment for build to start
             time.sleep(5) # Initial wait
             wait_time = 5
-            max_log_wait = 120 # Maximum total time to wait for logs in this step
+            max_log_wait = 150 # Increased max wait time for logs
             elapsed_log_wait = 0
             logs_fetched = False
             iframe_checked = False