Update app.py

app.py

@@ -6,9 +6,6 @@ from huggingface_hub import InferenceClient, HfApi, CommitOperationAdd
 # 1. Load Environment Token
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-if not HF_TOKEN:
-    raise ValueError("HF_TOKEN environment variable is not set. Please add it to your Space settings.")
-
 def load_migration_guide():
     """Loads the migration context from the external txt file."""
     try:
@@ -30,31 +27,47 @@ def extract_code_block(text: str) -> str:
         
     return ""
 
-def migrate_app(space_url):
+def migrate_app_generator(space_url):
     """
-    1. Fetches Space metadata and code using Env Token.
-    2. Sends code + migration guide to LLM.
-    3. Creates a PR on the Space.
+    Generator function that streams logs to a textbox and the final result to markdown.
     """
+    if not HF_TOKEN:
+        yield "❌ Error: HF_TOKEN environment variable is not set.", ""
+        return
+
     if not space_url:
-        raise gr.Error("Please enter a Space URL.")
+        yield "❌ Error: Please enter a Space URL.", ""
+        return
+
+    log_buffer = []
+
+    def log(message):
+        """Helper to append to log buffer and return joined string."""
+        print(message) # Console log
+        log_buffer.append(f"[{len(log_buffer)+1}] {message}")
+        return "\n".join(log_buffer)
 
-    # Setup Clients using the Environment Token
+    # 1. Initialization
+    yield log(f"🚀 Starting migration for: {space_url}"), ""
+
+    # Setup Clients
     api = HfApi(token=HF_TOKEN)
     client = InferenceClient(api_key=HF_TOKEN)
     
     # Verify Auth
     try:
         user = api.whoami()
-        print(f"Authenticated as {user['name']}")
+        yield log(f"✅ Authenticated as: {user['name']}"), ""
     except Exception as e:
-        raise gr.Error(f"Authentication failed using HF_TOKEN: {str(e)}")
+        yield log(f"❌ Authentication failed: {str(e)}"), ""
+        return
 
     # Parse Space ID
     space_id = space_url.replace("https://huggingface.co/spaces/", "").strip("/")
     
     try:
-        # Get README to update SDK version and find app file
+        # Get README
+        yield log(f"📥 Fetching README.md from {space_id}..."), ""
         readme_path = api.hf_hub_download(repo_id=space_id, filename="README.md", repo_type="space")
         with open(readme_path, "r", encoding="utf-8") as f:
             readme_content = f.read()
@@ -62,14 +75,18 @@ def migrate_app(space_url):
         # Determine python file name (default to app.py)
         app_file_match = re.search(r"app_file:\s*(.*)", readme_content)
         app_file_name = app_file_match.group(1).strip() if app_file_match else "app.py"
+        yield log(f"📄 Identified app file: {app_file_name}"), ""
         
         # Get Python Code
+        yield log(f"📥 Fetching {app_file_name}..."), ""
         code_path = api.hf_hub_download(repo_id=space_id, filename=app_file_name, repo_type="space")
         with open(code_path, "r", encoding="utf-8") as f:
             original_code = f.read()
+        yield log(f"✅ Code fetched successfully ({len(original_code)} chars)."), ""
             
     except Exception as e:
-        raise gr.Error(f"Error fetching files from {space_id}: {str(e)}")
+        yield log(f"❌ Error fetching files: {str(e)}"), ""
+        return
 
     # Prepare Prompt
     migration_guide = load_migration_guide()
@@ -92,7 +109,7 @@ def migrate_app(space_url):
     ]
 
     # Call Moonshot Model
-    gr.Info(f"Analysing {app_file_name} with moonshotai/Kimi-K2-Thinking...")
+    yield log("🧠 Analyzing code with moonshotai/Kimi-K2-Thinking (this may take 30-60s)..."), ""
     
     try:
         completion = client.chat.completions.create(
@@ -105,13 +122,20 @@ def migrate_app(space_url):
         migrated_code = extract_code_block(llm_response)
         
         if not migrated_code:
-            raise ValueError("LLM failed to generate valid Python code block.")
+            yield log("❌ LLM failed to generate valid Python code block."), ""
+            return
+            
+        yield log("✨ Code successfully migrated by LLM."), ""
             
     except Exception as e:
-        raise gr.Error(f"LLM Processing failed: {str(e)}")
+        yield log(f"❌ LLM Processing failed: {str(e)}"), ""
+        return
 
     # Prepare Commit
+    yield log("📦 preparing Pull Request..."), ""
+    
     # Update SDK version in README to 6.0.0
+    # Uses regex to replace sdk_version line
     new_readme_content = re.sub(
         r"sdk_version:.*", 
         "sdk_version: 6.0.0", 
@@ -141,9 +165,9 @@ def migrate_app(space_url):
             repo_type="space",
             create_pr=True
         )
-        return f"## ✅ Success!\n\nPull Request created: [**{commit_info.pr_url}**]({commit_info.pr_url})"
+        yield log("🎉 Done!"), f"## ✅ Success!\n\nPull Request created: [**{commit_info.pr_url}**]({commit_info.pr_url})"
     except Exception as e:
-        raise gr.Error(f"Failed to create Pull Request: {str(e)}")
+        yield log(f"❌ Failed to create Pull Request: {str(e)}"), ""
 
 # --- UI ---
 with gr.Blocks(title="Gradio 6 Auto-Migrator") as demo:
@@ -163,12 +187,21 @@ with gr.Blocks(title="Gradio 6 Auto-Migrator") as demo:
         )
         btn = gr.Button("Migrate Space", variant="primary", scale=1)
     
-    output_md = gr.Markdown(label="Status")
+    with gr.Row():
+        with gr.Column(scale=1):
+            log_output = gr.Textbox(
+                label="Execution Log", 
+                lines=10, 
+                interactive=False,
+                autoscroll=True
+            )
+        with gr.Column(scale=1):
+            result_output = gr.Markdown(label="Result")
 
     btn.click(
-        fn=migrate_app, 
+        fn=migrate_app_generator, 
         inputs=[space_input], 
-        outputs=output_md
+        outputs=[log_output, result_output]
     )
 
 if __name__ == "__main__":