Upload app.py with huggingface_hub

app.py

@@ -2,204 +2,122 @@ import gradio as gr
 import subprocess
 import os
 import json
-import tempfile
-from datetime import datetime
 
 # ============================================
 # TOOL DEFINITIONS
 # ============================================
 
 def tool_bash(command: str) -> str:
-    """Execute bash command and return output"""
     try:
-        result = subprocess.run(
-            command, 
-            shell=True, 
-            capture_output=True, 
-            text=True, 
-            timeout=30
-        )
-        output = result.stdout if result.stdout else result.stderr
-        return output[:5000] if output else "Command executed (no output)"
-    except subprocess.TimeoutExpired:
-        return "Error: Command timed out"
+        result = subprocess.run(command, shell=True, capture_output=True, text=True, timeout=30)
+        return result.stdout[:3000] if result.stdout else result.stderr[:3000] or "Done"
     except Exception as e:
         return f"Error: {str(e)}"
 
 def tool_write_file(path: str, content: str) -> str:
-    """Write content to a file"""
     try:
-        os.makedirs(os.path.dirname(path), exist_ok=True) if os.path.dirname(path) else None
-        with open(path, 'w', encoding='utf-8') as f:
+        with open(path, 'w') as f:
             f.write(content)
-        return f"File written successfully: {path}"
+        return f"Written to {path}"
     except Exception as e:
         return f"Error: {str(e)}"
 
 def tool_read_file(path: str) -> str:
-    """Read content from a file"""
     try:
-        with open(path, 'r', encoding='utf-8') as f:
-            content = f.read()
-        return content[:10000] if len(content) > 10000 else content
+        with open(path, 'r') as f:
+            return f.read()[:5000]
     except Exception as e:
         return f"Error: {str(e)}"
 
 def tool_list_files(directory: str = ".") -> str:
-    """List files in a directory"""
     try:
-        files = os.listdir(directory)
-        return "\n".join(files[:100])
+        return "\n".join(os.listdir(directory)[:50])
     except Exception as e:
         return f"Error: {str(e)}"
 
-def tool_delete_file(path: str) -> str:
-    """Delete a file"""
-    try:
-        os.remove(path)
-        return f"File deleted: {path}"
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Tool registry
 TOOLS = {
     "bash": tool_bash,
     "write_file": tool_write_file,
     "read_file": tool_read_file,
     "list_files": tool_list_files,
-    "delete_file": tool_delete_file,
 }
 
 # ============================================
-# AI AGENT LOGIC
+# AI LOGIC
 # ============================================
 
 from huggingface_hub import InferenceClient
 
-# Free inference client
 client = InferenceClient(model="Qwen/Qwen2.5-72B-Instruct")
 
-SYSTEM_PROMPT = """You are AI7, a helpful AI assistant with tool access.
-
-You can use these tools by responding with JSON:
-- bash: Run shell commands
-- write_file: Create files (path, content)
-- read_file: Read files (path)
-- list_files: List directory (directory)
-- delete_file: Remove files (path)
+SYSTEM = """You are AI7, a helpful AI assistant.
 
-To use a tool, reply with:
-TOOL: {"name": "tool_name", "arguments": {"arg": "value"}}
+Tools available (use JSON format to call):
+- bash: {"tool": "bash", "command": "ls"}
+- write_file: {"tool": "write_file", "path": "file.txt", "content": "text"}
+- read_file: {"tool": "read_file", "path": "file.txt"}
+- list_files: {"tool": "list_files", "dir": "."}
 
-After the tool result, continue helping the user. Be helpful and friendly."""
+When you need to use a tool, respond with ONLY the JSON, nothing else.
+Otherwise, respond normally to help the user."""
 
-def parse_tool_call(text: str):
-    """Parse tool call from model output"""
-    import re
-    match = re.search(r'TOOL:\s*(\{.*?\})', text, re.DOTALL)
-    if match:
-        try:
-            return json.loads(match.group(1))
-        except:
-            return None
-    return None
-
-def agent_chat(message, history):
-    """Main chat function"""
-    # Build messages list for the model
-    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+def chat(message, history):
+    messages = [{"role": "system", "content": SYSTEM}]
     
-    # Add conversation history (tuple format: (user, assistant))
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
+    for user, bot in history:
+        messages.append({"role": "user", "content": user})
+        messages.append({"role": "assistant", "content": bot})
     
     messages.append({"role": "user", "content": message})
     
     try:
-        # Get response from model
-        response = client.chat_completion(
-            messages=messages,
-            max_tokens=1024,
-            temperature=0.7
-        )
-        assistant_message = response.choices[0].message.content
-        
-        # Check for tool calls and execute
-        tool_call = parse_tool_call(assistant_message)
-        max_iterations = 5
+        response = client.chat_completion(messages=messages, max_tokens=1024)
+        reply = response.choices[0].message.content
         
-        while tool_call and max_iterations > 0:
-            tool_name = tool_call.get("name")
-            args = tool_call.get("arguments", {})
-            
-            if tool_name in TOOLS:
-                try:
-                    result = TOOLS[tool_name](**args)
-                except Exception as e:
-                    result = f"Error: {str(e)}"
+        # Check for tool call
+        try:
+            tool_data = json.loads(reply.strip())
+            if "tool" in tool_data:
+                tool_name = tool_data["tool"]
+                if tool_name == "bash":
+                    result = tool_bash(tool_data.get("command", ""))
+                elif tool_name == "write_file":
+                    result = tool_write_file(tool_data.get("path", ""), tool_data.get("content", ""))
+                elif tool_name == "read_file":
+                    result = tool_read_file(tool_data.get("path", ""))
+                elif tool_name == "list_files":
+                    result = tool_list_files(tool_data.get("dir", "."))
+                else:
+                    result = "Unknown tool"
                 
-                # Continue conversation with tool result
-                messages.append({"role": "assistant", "content": assistant_message})
+                # Get final response with tool result
+                messages.append({"role": "assistant", "content": reply})
                 messages.append({"role": "user", "content": f"Tool result:\n{result}"})
-                
-                response = client.chat_completion(
-                    messages=messages,
-                    max_tokens=1024,
-                    temperature=0.7
-                )
-                assistant_message = response.choices[0].message.content
-                tool_call = parse_tool_call(assistant_message)
-            else:
-                break
-            
-            max_iterations -= 1
+                response = client.chat_completion(messages=messages, max_tokens=1024)
+                return response.choices[0].message.content
+        except:
+            pass
         
-        return assistant_message
+        return reply
         
     except Exception as e:
-        return f"Error: {str(e)}\n\nNote: Using free Hugging Face inference. May have rate limits."
+        return f"Error: {str(e)}"
 
 # ============================================
-# GRADIO INTERFACE
+# INTERFACE
 # ============================================
 
-with gr.Blocks(title="AI7 Agent") as demo:
-    gr.Markdown("""
-    # AI7 - AI Agent with Tools
-    
-    An AI agent that can execute commands, read/write files, and help with tasks.
-    
-    **Tools:** bash, write_file, read_file, list_files, delete_file
-    
-    **Examples:**
-    - "Create a hello.py file"
-    - "List files in current directory"
-    - "What is 2+2?"
-    """)
-    
-    chatbot = gr.Chatbot(height=500)
+with gr.Blocks(title="AI7") as demo:
+    gr.Markdown("# AI7 - AI Agent\n\nAsk me anything. I can run commands and manage files.")
     
     with gr.Row():
-        msg = gr.Textbox(
-            placeholder="Type your message...",
-            show_label=False,
-            scale=9
-        )
-        submit = gr.Button("Send", scale=1, variant="primary")
-    
-    clear = gr.Button("Clear Chat")
+        inp = gr.Textbox(label="Message", scale=4)
+        btn = gr.Button("Send", variant="primary")
     
-    def respond(message, chat_history):
-        bot_message = agent_chat(message, chat_history)
-        chat_history.append((message, bot_message))
-        return "", chat_history
+    out = gr.Textbox(label="Response", lines=10)
     
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    submit.click(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: [], None, chatbot, queue=False)
+    btn.click(chat, [inp, gr.State([])], out)
+    inp.submit(chat, [inp, gr.State([])], out)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)