Update app.py

app.py

@@ -193,7 +193,8 @@ def run_agent(client, model_name, system_prompt_template, user_input_state, conf
     model_to_use = model_name
 
     try:
-        # *** FIX: Use a single Content object with role="user" and combined text in a Part ***
+        # Use a single Content object with role="user" and combined text in a Part
+        # The GenAI API for models like Gemini Flash Preview expects input roles to be 'user'.
         messages = [
             Content(role="user", parts=[Part(text=system_prompt + "\n\n" + user_message_content)])
         ]
@@ -441,26 +442,23 @@ 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 or if syntax errors occurred
-            if project_state['attempt_count'] == 0 and project_state['current_task'] == 'CODING - Initial Implementation':
-                # 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"""---
+            # Simplify the stubbing logic - just ensure these files exist in the state before CodeGen runs
+            if project_state['main_app_file'] not in project_state['files']:
+                print(f"Adding initial stub for {project_state['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"
+
+            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: 🐢
 sdk: {project_state['sdk_choice']}
@@ -477,7 +475,7 @@ This is an auto-generated HF Space.
 **Plan:**
 {project_state['plan']}
 """
-                    project_state['files']['README.md'] = readme_content
+                project_state['files']['README.md'] = readme_content
 
 
             step_successful = run_codegen(client, project_state, config)
@@ -499,10 +497,21 @@ This is an auto-generated HF Space.
                 create_repo(repo_id=project_state['repo_id'], token=oauth_token_token,
                             exist_ok=True, repo_type="space", space_sdk=project_state['sdk_choice'])
 
-                # Write and upload all files in project_state['files']
-                for fn, content in project_state['files'].items():
+                # *** FIX: Filter out any empty string keys before iterating ***
+                files_to_push = {
+                    fn: content
+                    for fn, content in project_state['files'].items()
+                    if fn and fn.strip() # Keep only non-empty, non-whitespace filenames
+                }
+                print(f"Attempting to push {len(files_to_push)} valid files...")
+
+
+                # Write and upload all valid files
+                for fn, content in files_to_push.items():
                      # Ensure directories exist for files like utils/data.py
-                     os.makedirs(os.path.dirname(fn), exist_ok=True)
+                     dirpath = os.path.dirname(fn)
+                     if dirpath: # Only create dir if filename has a path component
+                         os.makedirs(dirpath, exist_ok=True)
                      with open(fn, "w") as f:
                          f.write(content)
                      upload_file(
@@ -511,7 +520,7 @@ This is an auto-generated HF Space.
                          repo_type="space"
                      )
 
-                print(f"Pushed {len(project_state['files'])} files to {project_state['repo_id']}")
+                print(f"Pushed {len(files_to_push)} files to {project_state['repo_id']}")
                 project_state['status_message'] = f"Pushed code to HF Space **{project_state['repo_id']}**. Waiting for build..."
                 project_state['chat_history'].append({"role": "assistant", "content": project_state['status_message']})
                 project_state['current_task'] = 'LOGGING' # Move to fetching logs