Pro Edition: +restart_space, +check_status, +get_space_logs, +web_search, +example buttons, +better prompts

app.py

@@ -1,8 +1,11 @@
 """
-AI Coding Agent Crew - Fully Autonomous Edition
-================================================
+AI Coding Agent Crew - Pro Edition
+===================================
 "video downloader website ဆောက်ပေး" လို့ ပြောလိုက်ရုံနဲ့ အစအဆုံး လုပ်ပေးမယ်
 ဘာညွှန်ကြားချက်မှ မပါဘဲ ကိုယ့်ဟာကိုယ် ဆုံးဖြတ်ချက်ချပြီး လုပ်ဆောင်မယ်
+
+Tools: create_space, upload_file, set_secret, list_files, read_file,
+       delete_file, restart_space, check_status, get_space_logs, web_search
 """
 
 import os
@@ -18,7 +21,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 POLLINATIONS_URL = "https://text.pollinations.ai/openai/chat/completions"
-MAX_STEPS = 20
+MAX_STEPS = 25
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
 
 # ============================================================
@@ -43,6 +46,7 @@ logger.info(f"HF Username: {HF_USERNAME}")
 # ============================================================
 
 def tool_create_space(repo_id: str, sdk: str = "gradio") -> str:
+    """HF Space အသစ်ဆောက်မယ်"""
     from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
@@ -52,6 +56,7 @@ def tool_create_space(repo_id: str, sdk: str = "gradio") -> str:
         return f"ERROR: {e}"
 
 def tool_upload_file(repo_id: str, filename: str, content: str) -> str:
+    """HF Space မှာ file upload တင်မယ်"""
     from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
@@ -61,7 +66,7 @@ def tool_upload_file(repo_id: str, filename: str, content: str) -> str:
         api.upload_file(
             path_or_fileobj=tmp, path_in_repo=filename,
             repo_id=repo_id, repo_type="space",
-            commit_message=f"Add {filename}"
+            commit_message=f"Update {filename}"
         )
         os.unlink(tmp)
         return f"OK: Uploaded {filename} to {repo_id}"
@@ -69,6 +74,7 @@ def tool_upload_file(repo_id: str, filename: str, content: str) -> str:
         return f"ERROR: {e}"
 
 def tool_set_secret(repo_id: str, key: str, value: str) -> str:
+    """HF Space မှာ Secret သတ်မှတ်မယ်"""
     from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
@@ -78,6 +84,7 @@ def tool_set_secret(repo_id: str, key: str, value: str) -> str:
         return f"ERROR: {e}"
 
 def tool_list_files(repo_id: str) -> str:
+    """HF Space ရဲ့ files တွေ ကြည့်မယ်"""
     from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
@@ -87,6 +94,7 @@ def tool_list_files(repo_id: str) -> str:
         return f"ERROR: {e}"
 
 def tool_read_file(repo_id: str, filename: str) -> str:
+    """HF Space ကနေ file ဖတ်မယ်"""
     from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
@@ -100,6 +108,7 @@ def tool_read_file(repo_id: str, filename: str) -> str:
         return f"ERROR: {e}"
 
 def tool_delete_file(repo_id: str, filename: str) -> str:
+    """HF Space ကနေ file ဖျက်မယ်"""
     from huggingface_hub import HfApi
     api = HfApi(token=HF_TOKEN)
     try:
@@ -108,6 +117,77 @@ def tool_delete_file(repo_id: str, filename: str) -> str:
     except Exception as e:
         return f"ERROR: {e}"
 
+def tool_restart_space(repo_id: str) -> str:
+    """HF Space ကို restart လုပ်မယ် - deploy ပြီးရင် ဒါကို ခေါ်ပါ"""
+    from huggingface_hub import HfApi
+    api = HfApi(token=HF_TOKEN)
+    try:
+        api.restart_space(repo_id=repo_id)
+        return f"OK: Space '{repo_id}' restarted successfully"
+    except Exception as e:
+        return f"ERROR: {e}"
+
+def tool_check_status(repo_id: str) -> str:
+    """HF Space အလုပ်လုပ်နေလား စစ်မယ်"""
+    try:
+        from huggingface_hub import HfApi
+        api = HfApi(token=HF_TOKEN)
+        info = api.space_info(repo_id=repo_id)
+        stage = info.runtime.stage if info.runtime else "UNKNOWN"
+        hardware = info.runtime.hardware if info.runtime else "unknown"
+        
+        # Also try to access the space URL
+        space_url = f"https://{repo_id.replace('/', '-')}.hf.space"
+        try:
+            resp = req_lib.get(space_url, timeout=15)
+            http_status = resp.status_code
+        except:
+            http_status = "unreachable"
+        
+        return f"OK: Stage={stage}, Hardware={hardware}, HTTP={http_status}, URL={space_url}"
+    except Exception as e:
+        return f"ERROR: {e}"
+
+def tool_get_space_logs(repo_id: str) -> str:
+    """HF Space ရဲ့ runtime logs ဖတ်မယ် - Error ရှာဖို့ သုံးတယ်"""
+    try:
+        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+        resp = req_lib.get(
+            f"https://huggingface.co/api/spaces/{repo_id}",
+            headers=headers, timeout=30
+        )
+        if resp.status_code == 200:
+            data = resp.json()
+            runtime = data.get("runtime", {})
+            error_msg = runtime.get("errorMessage", None)
+            stage = runtime.get("stage", "unknown")
+            if error_msg:
+                return f"OK: Stage={stage}, Error={error_msg[:2000]}"
+            else:
+                return f"OK: Stage={stage}, No errors found. Space seems healthy."
+        else:
+            return f"ERROR: Could not fetch logs (status {resp.status_code})"
+    except Exception as e:
+        return f"ERROR: {e}"
+
+def tool_web_search(query: str) -> str:
+    """ဝဘ်မှာ ရှာမယ် - API examples, code snippets, documentation ရှာဖို့"""
+    try:
+        resp = req_lib.get(
+            f"https://text.pollinations.ai/",
+            params={"prompt": f"Search results for: {query}. Provide relevant URLs, code examples, and documentation links.", "model": "openai"},
+            timeout=30
+        )
+        if resp.status_code == 200:
+            text = resp.text[:3000]
+            return f"OK: Search results for '{query}':\n{text}"
+        else:
+            return f"ERROR: Search failed (status {resp.status_code})"
+    except Exception as e:
+        return f"ERROR: {e}"
+
+
+# Tool Registry
 TOOLS = {
     "create_space": tool_create_space,
     "upload_file": tool_upload_file,
@@ -115,6 +195,10 @@ TOOLS = {
     "list_files": tool_list_files,
     "read_file": tool_read_file,
     "delete_file": tool_delete_file,
+    "restart_space": tool_restart_space,
+    "check_status": tool_check_status,
+    "get_space_logs": tool_get_space_logs,
+    "web_search": tool_web_search,
 }
 
 # ============================================================
@@ -212,11 +296,67 @@ TOOL_DEFS = [
                 "required": ["repo_id", "filename"]
             }
         }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "restart_space",
+            "description": "Restart a Space after deploying files. Call this AFTER uploading all files.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "repo_id": {"type": "string", "description": "Space ID to restart"}
+                },
+                "required": ["repo_id"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "check_status",
+            "description": "Check if a Space is running, its stage, and HTTP status",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "repo_id": {"type": "string", "description": "Space ID"}
+                },
+                "required": ["repo_id"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_space_logs",
+            "description": "Get runtime error logs from a Space. Use this when a Space has RUNTIME_ERROR to diagnose issues.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "repo_id": {"type": "string", "description": "Space ID"}
+                },
+                "required": ["repo_id"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "web_search",
+            "description": "Search the web for API documentation, code examples, or how-to guides. Use when you need info about a library or API.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "query": {"type": "string", "description": "Search query"}
+                },
+                "required": ["query"]
+            }
+        }
     }
 ]
 
 # ============================================================
-# SYSTEM PROMPT
+# SYSTEM PROMPT - Agent ရဲ့ ဦးနှောက်
 # ============================================================
 
 SYSTEM_PROMPT = f"""You are a fully autonomous AI coding agent. You make ALL decisions yourself and execute tasks from start to finish WITHOUT asking the user for more instructions.
@@ -229,7 +369,7 @@ When a user asks you to build something, you must:
 1. AUTOMATICALLY plan all steps needed
 2. Execute each step by calling the appropriate tool
 3. After each tool result, decide and execute the next step
-4. Fix errors automatically - never give up
+4. Fix errors automatically by reading logs and retrying
 5. Continue until the task is FULLY COMPLETE
 6. Give the user the final Space URL
 
@@ -238,26 +378,59 @@ When a user asks you to build something, you must:
 - NEVER say "please provide..." or "could you tell me..." - JUST DO IT
 - ALWAYS upload BOTH app.py AND requirements.txt for any new Space
 - Write COMPLETE, WORKING Python code - NO placeholders, NO "...", NO "# your code here"
-- Use gradio>=5.0.0 for all web UIs
-- After ALL files are uploaded, provide the Space URL
+- After ALL files are uploaded, restart the Space and check status
+- If Space has RUNTIME_ERROR, read logs with get_space_logs, fix the code, re-upload
 - Reply in the SAME LANGUAGE as the user
 - If you get an error, analyze it and FIX it - try a different approach
 
-## STEP-BY-STEP for building a web app:
+## DEPLOYMENT WORKFLOW:
 1. create_space(repo_id="{HF_USERNAME}/app-name", sdk="gradio")
 2. upload_file(repo_id=..., filename="app.py", content=COMPLETE_CODE)
 3. upload_file(repo_id=..., filename="requirements.txt", content="gradio>=5.0.0\\n...")
-4. list_files(repo_id=...) to verify
-5. Provide Space URL: https://huggingface.co/spaces/{{repo_id}}
+4. restart_space(repo_id=...) to activate
+5. check_status(repo_id=...) to verify
+6. If error: get_space_logs(repo_id=...) → read error → fix code → re-upload → restart
+7. Provide Space URL: https://huggingface.co/spaces/{{repo_id}}
 
-## CODE GUIDELINES:
+## IMPORTANT CODE GUIDELINES:
 - Always: import gradio as gr
 - Use gr.Blocks() or gr.Interface()
 - Always include demo.launch() at the end
 - Handle errors in the code with try/except
 - Include proper imports at the top
+- For Gradio 6.x: do NOT use type="messages" in Chatbot, do NOT pass css to Blocks (pass to launch)
+- For Gradio 6.x: Chatbot uses messages format by default (list of dicts with role/content)
+- Use gr.ChatInterface for simple chat apps
+
+## CODE TEMPLATES:
+
+Simple Chat App:
+```python
+import gradio as gr
+
+def chat(message, history):
+    return f"Echo: {{message}}"
+
+demo = gr.ChatInterface(fn=chat, title="My Chat App")
+demo.launch()
+```
+
+Blocks App with UI:
+```python
+import gradio as gr
+
+with gr.Blocks(title="My App") as demo:
+    gr.Markdown("# My App")
+    input_text = gr.Textbox(label="Input")
+    output_text = gr.Textbox(label="Output")
+    btn = gr.Button("Submit")
+    btn.click(fn=lambda x: x.upper(), inputs=input_text, outputs=output_text)
+
+demo.launch()
+```
 
-Remember: You are AUTONOMOUS. Make decisions. Take action. Complete the task."""
+Remember: You are AUTONOMOUS. Make decisions. Take action. Complete the task.
+After deploying, ALWAYS restart the Space and verify it works. If it fails, read logs and fix it."""
 
 # ============================================================
 # AI CALLING
@@ -281,23 +454,17 @@ def call_ai(messages, retries=2):
             else:
                 logger.warning(f"AI status {resp.status_code} (attempt {attempt+1})")
                 if attempt < retries:
-                    import time
-                    time.sleep(3)
-                    continue
+                    import time; time.sleep(3); continue
                 return {"content": f"AI API error ({resp.status_code})."}
         except req_lib.Timeout:
             logger.warning(f"AI timeout (attempt {attempt+1})")
             if attempt < retries:
-                import time
-                time.sleep(5)
-                continue
+                import time; time.sleep(5); continue
             return {"content": "AI request timed out."}
         except Exception as e:
             logger.error(f"call_ai error: {e}")
             if attempt < retries:
-                import time
-                time.sleep(3)
-                continue
+                import time; time.sleep(3); continue
             return {"content": f"AI error: {e}"}
     return {"content": "AI unavailable."}
 
@@ -455,7 +622,6 @@ def history_to_messages(history):
             continue
 
         content = item.get("content", "")
-        # Gradio 6 messages format: content can be a list
         if isinstance(content, list):
             texts = []
             for sub in content:
@@ -469,7 +635,6 @@ def history_to_messages(history):
         if not content:
             continue
 
-        # Clean progress UI from assistant messages
         if role == "assistant":
             content = _clean_bot_msg(content)
 
@@ -481,10 +646,9 @@ def history_to_messages(history):
 
 def _clean_bot_msg(text):
     """Remove progress UI elements from bot messages for AI context"""
-    # Remove progress blocks
     text = re.sub(r'🚀 လုပ်ဆောင်နေပါပြီ\.\.\.', '', text)
     text = re.sub(r'📋 လုပ်ဆောင်ချက်များ:.*?(?=\n\n|\Z)', '', text, flags=re.DOTALL)
-    text = re.sub(r'⏳ Step \d+/\d+ လုပ်ဆောင်နေဆဲ\.\.\.', '', text)
+    text = re.sub(r'⏳ Step \d+/\d+.*?\n', '', text)
     text = re.sub(r'✅ Step \d+:.*?\n', '', text)
     text = re.sub(r'🔄 Step \d+:.*?\n', '', text)
     text = re.sub(r'\n{3,}', '\n\n', text)
@@ -495,9 +659,8 @@ def _clean_bot_msg(text):
 # ============================================================
 
 def run_agent(user_message, history):
-    """Fully autonomous agent - yields (history, response_text) for Gradio"""
+    """Fully autonomous agent - yields response_text for Gradio"""
 
-    # Build messages for AI
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     messages.extend(history_to_messages(history))
     messages.append({"role": "user", "content": user_message})
@@ -512,7 +675,6 @@ def run_agent(user_message, history):
 
         # 2. No tool call = final response
         if not tool_call:
-            # Check if model is just planning without acting
             if text_content and _is_planning_text(text_content):
                 messages.append({"role": "assistant", "content": text_content})
                 messages.append({
@@ -524,7 +686,6 @@ def run_agent(user_message, history):
                 yield progress
                 continue
 
-            # Final answer
             if text_content:
                 final = _format_final(steps_done, text_content)
             else:
@@ -552,24 +713,29 @@ def run_agent(user_message, history):
             messages.append({"role": "assistant", "content": f"Calling {tool_name}"})
 
         if result.startswith("ERROR"):
+            # If it's a Space error, suggest reading logs
+            extra = ""
+            if "space" in tool_name.lower() or "upload" in tool_name.lower():
+                repo_id = tool_args.get("repo_id", "")
+                if repo_id:
+                    extra = f"\n\nTry: get_space_logs(repo_id='{repo_id}') to see the error details."
             messages.append({
                 "role": "user",
-                "content": f"❌ Tool FAILED: {result}\n\nAnalyze the error and try a different approach. Fix it and retry."
+                "content": f"❌ Tool FAILED: {result}{extra}\n\nAnalyze the error and try a different approach. Fix it and retry."
             })
         else:
             done = _check_task_complete(tool_name, result, steps_done)
             if done:
                 messages.append({
                     "role": "user",
-                    "content": f"✅ Result: {result}\n\nAll files uploaded. Provide the final Space URL and summary. Do NOT call more tools."
+                    "content": f"✅ Result: {result}\n\nAll files uploaded and Space restarted. Provide the final Space URL and summary to the user. Do NOT call more tools."
                 })
             else:
                 messages.append({
                     "role": "user",
-                    "content": f"✅ Result: {result}\n\nContinue with the next step, or if done, provide the Space URL."
+                    "content": f"✅ Result: {result}\n\nContinue with the next step. If all files are uploaded, restart the Space and check status."
                 })
 
-    # Max steps
     yield _format_final(steps_done, "⚠️ အများဆုံး အဆင့်ရောက်ပါပြီ။ ဆက်လုပ်လိုပါက ထပ်မေးပါ။")
 
 
@@ -589,24 +755,40 @@ def _describe_step(tool_name, tool_args, result):
     if tool_name == "create_space":
         return f"Space ဆောက်တယ်: {tool_args.get('repo_id', '?')} ({tool_args.get('sdk', 'gradio')})"
     elif tool_name == "upload_file":
-        return f"File တင်တယ်: {tool_args.get('filename', '?')} → {tool_args.get('repo_id', '?')} ({len(tool_args.get('content', ''))} chars)"
+        return f"File တင်တယ်: {tool_args.get('filename', '?')} ({len(tool_args.get('content', ''))} chars)"
     elif tool_name == "set_secret":
-        return f"Secret သတ်မှတ်တယု: {tool_args.get('key', '?')} → {tool_args.get('repo_id', '?')}"
+        return f"Secret သတ်မှတ်တယု: {tool_args.get('key', '?')}"
     elif tool_name == "list_files":
         return f"Files ကြည့်တယု: {tool_args.get('repo_id', '?')}"
     elif tool_name == "read_file":
         return f"File ဖတ်တယု: {tool_args.get('filename', '?')}"
     elif tool_name == "delete_file":
         return f"File ဖျက်တယု: {tool_args.get('filename', '?')}"
+    elif tool_name == "restart_space":
+        return f"Space restart လုပ်တယု: {tool_args.get('repo_id', '?')}"
+    elif tool_name == "check_status":
+        return f"Status စစ်တယု: {tool_args.get('repo_id', '?')}"
+    elif tool_name == "get_space_logs":
+        return f"Logs ဖတ်တယု: {tool_args.get('repo_id', '?')}"
+    elif tool_name == "web_search":
+        return f"Web ရှာတယု: {tool_args.get('query', '?')[:50]}"
     return f"{tool_name}"
 
 
 def _check_task_complete(tool_name, result, steps_done):
     created = any("Space ဆောက်တယု" in s for s in steps_done)
     uploaded_app = any("app.py" in s for s in steps_done)
-    if tool_name == "list_files" and created and uploaded_app:
+    restarted = any("restart" in s.lower() for s in steps_done)
+    checked = any("Status စစ်တယု" in s or "Logs ဖတ်တယု" in s for s in steps_done)
+    
+    # If we've restarted and checked, we're done
+    if tool_name == "check_status" and restarted and uploaded_app:
+        return True
+    # If we've read logs after a restart, also done
+    if tool_name == "get_space_logs" and restarted and uploaded_app:
         return True
-    if tool_name == "upload_file" and "requirements.txt" in result and uploaded_app:
+    # If we restarted after uploading both files
+    if tool_name == "restart_space" and created and uploaded_app:
         return True
     return False
 
@@ -633,9 +815,18 @@ def _format_final(steps_done, final_text):
 
 
 # ============================================================
-# GRADIO UI - Compatible with Gradio 4.x, 5.x, 6.x
+# GRADIO UI
 # ============================================================
 
+EXAMPLES = [
+    "video downloader website ဆောက်ပေး",
+    "image generator app လုပ်ပေး",
+    "chatbot website ဆောက်ပေး",
+    "text to speech app လုပ်ပေး",
+    "PDF summarizer app ဆောက်ပေး",
+    "မင်းရဲ့ Space တွေ ဘာတွေရှိလဲ",
+]
+
 with gr.Blocks(title="AI Coding Agent Crew") as demo:
     gr.Markdown("# 🤖 AI Coding Agent Crew")
     gr.Markdown("""**ကိုယ့်ဟာကိုယ် ဆုံးဖြတ်ချက်ချပြီး ကိုယ်ဘာသာ လုပ်ဆောင်နိုင်တဲ့ Agent**
@@ -643,17 +834,30 @@ with gr.Blocks(title="AI Coding Agent Crew") as demo:
 လုပ်ဆောင်ချက်တစ်ခု ပြောလိုက်ရုံပါ - Agent က အစအဆုံး ကိုယ့်ဟာကိုယ် လုပ်ပေးမှာ:
 • "video downloader website ဆောက်ပေး"
 • "image generator app လုပ်ပေး"
-• "chatbot website ဆောက်ပေး" """)
+• "chatbot website ဆောက်ပေး"
+• "PDF summarizer app ဆောက်ပေး"
+
+🔧 **Tools**: Space ဆောက်, File တင်, Secret သတ်မှတ်, Restart, Status စစ်, Logs ဖတ်, Web ရှာ""")
 
     chatbot = gr.Chatbot(height=500)
-    msg_input = gr.Textbox(
-        placeholder="ဘာလုပ်ချင်လဲ ရိုက်ထည့်ပါ...",
-        show_label=False,
-        scale=4,
-    )
+    
     with gr.Row():
+        msg_input = gr.Textbox(
+            placeholder="ဘာလုပ်ချင်လဲ ရိုက်ထည့်ပါ...",
+            show_label=False,
+            scale=4,
+        )
         send_btn = gr.Button("ပို့ရန်", variant="primary", scale=1)
+    
+    with gr.Row():
         clear_btn = gr.ClearButton([msg_input, chatbot], scale=1)
+    
+    # Example buttons
+    gr.Examples(
+        examples=EXAMPLES,
+        inputs=msg_input,
+        label="နမူနာ ခလုတ်တွေ (နှိပ်လိုက်ရုံပါ)"
+    )
 
     def user_submit(user_message, history):
         """Add user message to chat (messages format)"""
@@ -667,7 +871,6 @@ with gr.Blocks(title="AI Coding Agent Crew") as demo:
             yield history
             return
 
-        # Get the last user message text
         last_msg = history[-1]
         user_message = ""
         content = last_msg.get("content", "") if isinstance(last_msg, dict) else ""
@@ -683,17 +886,13 @@ with gr.Blocks(title="AI Coding Agent Crew") as demo:
             yield history
             return
 
-        # Previous messages (before the last user message)
         prev_history = history[:-1]
-
-        # Add placeholder assistant message
         history.append({"role": "assistant", "content": "🚀 စတင်လုပ်ဆောင်နေပါပြီ..."})
 
         for response_text in run_agent(user_message, prev_history):
             history[-1] = {"role": "assistant", "content": response_text}
             yield history
 
-    # Handle submit
     msg_input.submit(
         user_submit,
         [msg_input, chatbot],