Update app.py from anycoder

app.py

@@ -4,93 +4,142 @@ import time
 from datetime import datetime
 import tempfile
 import os
+import traceback
+
+def safe_write_temp_file(content: str, suffix: str = ".txt") -> str | None:
+    """
+    Safely writes content to a temporary file.
+    Returns the file path if successful, None otherwise.
+    """
+    try:
+        # Create a temp file that persists after closing (delete=False)
+        # We use a try/finally block to ensure we close the file handle
+        # before returning the path, preventing OS locking errors.
+        temp_file = tempfile.NamedTemporaryFile(mode='w', delete=False, suffix=suffix)
+        temp_file.write(content)
+        temp_file.close()
+        return temp_file.name
+    except IOError as e:
+        print(f"File System Error: {e}")
+        return None
+    except Exception as e:
+        print(f"Unexpected File Error: {e}")
+        return None
 
 def simulate_processing(
     content: str, 
     operation: str, 
     iteration: int, 
     creativity_level: float
-) -> tuple[str, int, str, str | None]:
+) -> tuple[str, int, str, gr.DownloadButton | None]:
     """
     Simulates a Review, Revise, Refactor, and Iterate workflow.
-    Includes simulated 'AI' parameters for a studio feel.
+    Includes comprehensive error handling.
     """
-    if not content.strip():
-        return "", 0, "⚠️ **Please enter some text or code to begin.**", None
+    # 1. Initial Validation (Double check, though validator handles the first pass)
+    if content is None:
+        return "", 0, "⚠️ **Error:** No input detected.", None
 
-    # Simulate processing time based on creativity level
-    delay = 0.3 + (creativity_level * 0.5)
-    time.sleep(delay)
-    
-    result = content
-    status_msg = ""
-    download_filename = None
-    
-    # Generate a timestamp for logs
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-
-    if operation == "Review":
-        status_msg = f"""
-        ### ✅ Review Complete
-        - **Status:** Pass
-        - **Time:** {timestamp}
-        - **Analysis:** Syntax looks valid. 
-        - **Suggestion:** Consider adding docstrings and type hints for better maintainability.
-        """
-        result = f"# REVIEW LOG [{timestamp}]\n# Status: Pass\n# Note: Code structure is solid.\n\n{content}"
-        download_filename = "review_log.py"
-
-    elif operation == "Revise":
-        status_msg = """
-        ### ✍️ Revision Complete
-        - **Tone:** Professional
-        - **Grammar:** Corrected
-        - **Flow:** Optimized
-        """
-        # Simple text simulation: capitalize and clean up
-        sentences = content.split('. ')
-        revised = '. '.join([s.strip().capitalize() for s in sentences if s.strip()])
-        result = revised if revised else content
-        download_filename = "revised_text.txt"
-
-    elif operation == "Refactor":
-        status_msg = """
-        ### ⚙️ Refactoring Complete
-        - **Optimization:** Applied
-        - **Complexity:** Reduced
-        - **Style:** PEP8 Compliant
-        """
-        # Simple code simulation
-        result = content.replace("var ", "let ").replace("  ", " ")
-        result = f"# REFACTORED VERSION [{timestamp}]\n# Optimized for performance\n{result}"
-        download_filename = "refactored_code.py"
-
-    elif operation == "Iterate":
-        iteration += 1
-        status_msg = f"""
-        ### 🚀 Iteration {iteration} Generated
-        - **Creativity Level:** {creativity_level:.2f}
-        - **Action:** Created new version based on feedback.
-        """
-        result = f"# --- ITERATION {iteration} ---\n# Generated at {timestamp}\n# Creativity: {creativity_level}\n{content}"
-        download_filename = f"iteration_{iteration}.py"
+    try:
+        # Simulate processing time based on creativity level
+        # Capped to prevent excessive waiting
+        delay = min(0.3 + (creativity_level * 0.5), 2.0)
+        time.sleep(delay)
+        
+        result = content
+        status_msg = ""
+        download_filename = None
+        
+        # Generate a timestamp for logs
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
-    # Create a temporary file for download
-    temp_file = None
-    if download_filename and result:
-        with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".py") as f:
-            f.write(result)
-            temp_file = f.name
+        if operation == "Review":
+            status_msg = f"""
+            ### ✅ Review Complete
+            - **Status:** Pass
+            - **Time:** {timestamp}
+            - **Analysis:** Syntax looks valid. 
+            - **Suggestion:** Consider adding docstrings and type hints for better maintainability.
+            """
+            result = f"# REVIEW LOG [{timestamp}]\n# Status: Pass\n# Note: Code structure is solid.\n\n{content}"
+            download_filename = "review_log.py"
 
-    return result, iteration, status_msg, temp_file
+        elif operation == "Revise":
+            status_msg = """
+            ### ✍️ Revision Complete
+            - **Tone:** Professional
+            - **Grammar:** Corrected
+            - **Flow:** Optimized
+            """
+            # Logic: Ensure we handle string operations safely
+            sentences = content.split('. ')
+            revised = '. '.join([s.strip().capitalize() for s in sentences if s.strip()])
+            result = revised if revised else content
+            download_filename = "revised_text.txt"
 
-def create_download_file(content: str) -> str:
-    """Helper to create a downloadable file object."""
-    if not content:
-        return None
-    with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
-        f.write(content)
-        return f.name
+        elif operation == "Refactor":
+            status_msg = """
+            ### ⚙️ Refactoring Complete
+            - **Optimization:** Applied
+            - **Complexity:** Reduced
+            - **Style:** PEP8 Compliant
+            """
+            # Logic: Safe string replacement
+            result = content.replace("var ", "let ").replace("  ", " ")
+            result = f"# REFACTORED VERSION [{timestamp}]\n# Optimized for performance\n{result}"
+            download_filename = "refactored_code.py"
+
+        elif operation == "Iterate":
+            iteration += 1
+            status_msg = f"""
+            ### 🚀 Iteration {iteration} Generated
+            - **Creativity Level:** {creativity_level:.2f}
+            - **Action:** Created new version based on feedback.
+            """
+            result = f"# --- ITERATION {iteration} ---\n# Generated at {timestamp}\n# Creativity: {creativity_level}\n{content}"
+            download_filename = f"iteration_{iteration}.py"
+
+        # 2. File Generation with Error Handling
+        temp_file_path = None
+        if result:
+            temp_file_path = safe_write_temp_file(result, ".txt")
+            
+            if not temp_file_path:
+                # If file write failed, warn the user but don't crash the whole app
+                status_msg += "\n\n⚠️ **Warning:** Could not generate download file due to a system error."
+
+        # 3. Return success state
+        # We return an updated DownloadButton component
+        return result, iteration, status_msg, gr.DownloadButton(value=temp_file_path, visible=True)
+
+    except Exception as e:
+        # 4. Global Exception Catch
+        # This catches any unexpected crashes in the logic above
+        error_trace = traceback.format_exc()
+        print(f"Critical Error in Processing: {error_trace}")
+        
+        error_msg = f"""
+        ### 🚨 Critical Error
+        The workflow encountered an unexpected error.
+        
+        **Error Details:** `{str(e)}`
+        
+        *Your input has been preserved. Please try again or check your input format.*
+        """
+        
+        # Return error state (no download button, preserve input)
+        return content, iteration, error_msg, gr.DownloadButton(visible=False, value=None)
+
+def validate_inputs(content: str, operation: str):
+    """
+    Gradio 6 Validator function.
+    Runs immediately on the client side before queuing.
+    """
+    if not content or not content.strip():
+        return [
+            gr.validate(False, "Input cannot be empty. Please enter text or code to proceed.")
+        ]
+    return [gr.validate(True)] # Valid
 
 # Gradio 6 Application
 # NO parameters in gr.Blocks() constructor!
@@ -177,37 +226,43 @@ with gr.Blocks() as demo:
                 interactive=True
             )
             
+            # Initialize download button as hidden
             download_btn = gr.DownloadButton(
                 label="📥 Download Result",
                 variant="secondary",
-                visible=False
+                visible=False,
+                size="lg"
             )
 
     # Event Listeners using Gradio 6 syntax
-    def process_wrapper(content, op, iter_count, creat):
-        res, new_iter, status, temp_path = simulate_processing(content, op, iter_count, creat)
-        return res, new_iter, status, gr.DownloadButton(visible=True, value=temp_path)
-
+    
+    # 1. Main Execution Event
+    # We attach the validator here to check inputs before running the heavy function
     run_btn.click(
-        fn=process_wrapper,
+        fn=simulate_processing,
+        validator=validate_inputs,
         inputs=[input_code, operation_mode, iteration_state, creativity],
         outputs=[output_code, iteration_state, status_output, download_btn],
         api_visibility="public"
     )
 
+    # 2. Clear Button Event
     clear_btn.click(
-        fn=lambda: ("", 0, "*Ready to process...*", gr.DownloadButton(visible=False, value=None), "Review", 0.5),
+        fn=lambda: (
+            "",  # input_code
+            0,   # iteration_state
+            "*Ready to process...*", # status_output
+            gr.DownloadButton(visible=False, value=None), # download_btn
+            "Review", # operation_mode
+            0.5 # creativity
+        ),
         inputs=None,
         outputs=[input_code, iteration_state, status_output, download_btn, operation_mode, creativity],
         api_visibility="private"
     )
 
-    # Allow downloading directly from the output code box if modified manually
-    output_code.change(
-        fn=lambda x: gr.DownloadButton(visible=True, value=create_download_file(x)) if x else gr.DownloadButton(visible=False),
-        inputs=[output_code],
-        outputs=[download_btn]
-    )
+    # NOTE: Removed output_code.change listener to prevent file generation loops/locks.
+    # The download button is now exclusively updated by the Execute Workflow button.
 
 # Gradio 6 Launch Configuration
 # ALL app-level parameters (theme, css, etc.) go here!
@@ -235,7 +290,6 @@ demo.launch(
         {"label": "GitHub", "url": "https://github.com/gradio-app/gradio"},
         {"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}
     ],
-    # Add a custom footer message
     head="""
     <style>
         .gradio-container {