Update deployment result link and download flow

app.py

@@ -47,6 +47,9 @@ class SpaceCreatorApp:
             try:
                 if os.path.exists(file_path):
                     os.remove(file_path)
+                parent_dir = os.path.dirname(file_path)
+                if parent_dir and os.path.isdir(parent_dir) and not os.listdir(parent_dir):
+                    os.rmdir(parent_dir)
             except:
                 pass
         self.temp_files = []
@@ -92,14 +95,27 @@ class SpaceCreatorApp:
     def process_step2(self, hf_token, space_name, deploy_mode):
         """Step 2: Deploy to Hugging Face"""
         try:
-            if deploy_mode != "Deploy to Hugging Face Space":
+            deploy_mode = deploy_mode or ""
+            download_only = deploy_mode in ("📥 Download Only", "Download Only")
+            deploy_space = deploy_mode in ("🚀 Deploy to Hugging Face Space", "Deploy to Hugging Face Space")
+
+            if download_only:
+                zip_bytes = create_zip_archive(self.generated_files)
+                zip_path = save_individual_file("gradio_app_full_package.zip", zip_bytes, binary=True)
+                self.temp_files.append(zip_path)
+
+                status = self._create_status_message("success",
+                    "Download package ready! Use the ZIP download button below.")
+                return zip_path, status, 3, ""
+
+            if not deploy_space:
                 status = self._create_status_message("info", "Skipping deployment as requested.")
-                return None, status, 2
+                return None, status, 2, ""
             
             # Validate deployment inputs
             validation_errors = validate_inputs(None, hf_token, space_name, "Deploy")
             if validation_errors:
-                return None, self._format_errors(validation_errors), 2
+                return None, self._format_errors(validation_errors), 2, ""
             
             # Deploy to Hugging Face
             deploy_url = deploy_to_space(hf_token, space_name, self.generated_files)
@@ -111,25 +127,12 @@ class SpaceCreatorApp:
             
             status = self._create_status_message("success", 
                 f"Deployment successful! Space URL: {deploy_url}")
-            return zip_path, status, 3
-            
+            link_markdown = f"[Open your newly deployed Space]({deploy_url})"
+            return zip_path, status, 3, link_markdown
+        
         except Exception as e:
             error_msg = self._create_status_message("error", f"Deployment failed: {str(e)}")
-            return None, error_msg, 2
-    
-    def _extract_code(self, input_text, input_file):
-        """Extract code from text or file"""
-        code_content = ""
-        if input_file is not None:
-            file_content = read_file_content(input_file)
-            if input_file.name.endswith('.ipynb') if hasattr(input_file, 'name') else False:
-                with tempfile.NamedTemporaryFile(mode='w', suffix='.ipynb', delete=False, encoding='utf-8') as tmp:
-                    tmp.write(file_content)
-                    tmp_path = tmp.name
-                try:
-                    code_content = parse_notebook(tmp_path)
-                finally:
-                    os.unlink(tmp_path)
+            return None, error_msg, 2, ""
             else:
                 code_content = file_content
         elif input_text.strip():
@@ -1136,7 +1139,7 @@ def create_app():
         deploy_btn.click(
             fn=app.process_step2,
             inputs=[hf_token, space_name, deploy_mode],
-            outputs=[full_package_download, status_output, current_step]
+            outputs=[full_package_download, status_output, current_step, deployment_status]
         ).then(
             fn=lambda step: (gr.update(visible=step==1), gr.update(visible=step==2), gr.update(visible=step==3)),
             inputs=[current_step],

utils/converter.py

@@ -275,11 +275,12 @@ All dependencies are listed in requirements.txt
 def create_basic_app_template(input_code: str) -> str:
     """Create a basic, guaranteed-valid app.py template"""
     code_snippet = input_code[:1000]
+    commented_code = '\n'.join(f'# {line}' if line.strip() else '#' for line in code_snippet.splitlines())
     
     return f'''import gradio as gr
 
 # Original code snippet
-{code_snippet}
+{commented_code}
 
 # Gradio interface
 def process_input(input_text):
@@ -317,7 +318,7 @@ with gr.Blocks(title="Generated Application", theme=gr.themes.Soft()) as demo:
 
 if __name__ == "__main__":
     demo.launch(share=False)
-'''
+'''}]}
 
 
 def create_basic_template(input_code: str) -> Dict[str, str]:

utils/file_handler.py

@@ -39,6 +39,8 @@ def read_file_content(file_obj) -> str:
         elif isinstance(file_obj, (str, Path)):
             path = Path(file_obj)
             if path.exists():
+                if path.stat().st_size > 10 * 1024 * 1024:
+                    raise ValueError("Uploaded file exceeds the 10MB limit.")
                 return path.read_text(encoding='utf-8', errors='ignore')
     except Exception as e:
         print(f"Error reading file: {e}")

utils/validator.py

@@ -28,15 +28,19 @@ def validate_inputs(
             errors.append("Invalid Groq API Key format")
     
     # Hugging Face Token validation for deployment
-    if mode == "Deploy" and hf_token:
-        if not hf_token.startswith('hf_'):
+    if mode == "Deploy":
+        if not hf_token:
+            errors.append("Hugging Face Token is required for deployment")
+        elif not hf_token.startswith('hf_'):
             errors.append("Hugging Face token should start with 'hf_'")
         elif len(hf_token) < 10:
             errors.append("Invalid Hugging Face token format")
     
     # Space Name validation for deployment
-    if mode == "Deploy" and space_name:
-        if len(space_name) < 3 or len(space_name) > 30:
+    if mode == "Deploy":
+        if not space_name:
+            errors.append("Space name is required for deployment")
+        elif len(space_name) < 3 or len(space_name) > 30:
             errors.append("Space name must be 3-30 characters")
         elif not re.match(r'^[a-z0-9-]+$', space_name):
             errors.append("Space name can only contain lowercase letters, numbers, and hyphens")