Upload 18 files

app.py

@@ -2,7 +2,8 @@
 import os
 import base64
 import io
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel 
 from PIL import Image
@@ -27,18 +28,65 @@ class UserRequest(BaseModel):
 class CodeRequest(BaseModel):
     user_input: str
 
+def get_workspace_files():
+    workspace = code_agent.cfg.get("work_dir", "./workspace")
+    if not os.path.exists(workspace):
+        return set()
+    return set(os.listdir(workspace))
+
 @app.post("/code/chat")
 def handle_code_chat(request: CodeRequest):
     try:
+        # Detect existing files
+        files_before = get_workspace_files()
+        
+        # Run agent
         response = code_agent.chat_loop(request.user_input)
+        
+        # Detect new files
+        files_after = get_workspace_files()
+        new_files = files_after - files_before
+        
+        generated_images = []
+        workspace = code_agent.cfg.get("work_dir", "./workspace")
+        
+        for filename in new_files:
+            if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
+                file_path = os.path.join(workspace, filename)
+                try:
+                    with open(file_path, "rb") as img_file:
+                        b64_data = base64.b64encode(img_file.read()).decode('utf-8')
+                        generated_images.append({
+                            "filename": filename,
+                            "b64": b64_data
+                        })
+                except Exception as e:
+                    print(f"Error reading generated image {filename}: {e}")
+
         return {
             "success": True,
-            "bot_response": response
+            "bot_response": response,
+            "generated_images": generated_images
         }
     except Exception as e:
         print(f"Erro crítico no endpoint /code/chat: {e}")
         return {"success": False, "error": str(e)}
 
+@app.get("/download/{filename}")
+def download_file(filename: str):
+    # Sanitize and check existence in workspace
+    workspace = code_agent.cfg.get("work_dir", "./workspace")
+    file_path = os.path.join(workspace, filename)
+    
+    # Basic path traversal protection
+    if not os.path.abspath(file_path).startswith(os.path.abspath(workspace)):
+         raise HTTPException(status_code=403, detail="Access denied")
+         
+    if os.path.exists(file_path) and os.path.isfile(file_path):
+        return FileResponse(file_path, filename=filename)
+    else:
+        raise HTTPException(status_code=404, detail="File not found")
+
 @app.post("/chat")
 def handle_chat(request: UserRequest):
     try:

code_jade/config.json

@@ -1,5 +1,5 @@
 {
-  "groq_model": "moonshotai/kimi-k2-instruct-0905",
+  "groq_model": "llama3-70b-8192",
   "max_context": 20,
   "safe_mode": false,
   "work_dir": "./workspace"

code_jade/core.py

@@ -12,7 +12,7 @@ class CodeJadeAgent:
                 self.cfg = json.load(f)
         except FileNotFoundError:
             # Fallback se não achar, mas idealmente deve existir
-            self.cfg = {"groq_model": "moonshotai/kimi-k2-instruct-0905", "safe_mode": True, "work_dir": "./workspace", "max_context": 20}
+            self.cfg = {"groq_model": "llama3-70b-8192", "safe_mode": True, "work_dir": "./workspace", "max_context": 20}
 
         self.client = Groq(api_key=self._get_api_key())
         self.tools = ToolManager(safe_mode=self.cfg.get("safe_mode", True), work_dir=self.cfg.get("work_dir", "."))
@@ -169,6 +169,9 @@ CodeJade: "Arquivo criado. Quer que eu execute?"
         
         final_response = ""
 
+        last_tool_call = None
+        consecutive_tool_failures = 0
+
         while turn < max_turns:
             # 2. Chama o modelo
             response = self._chat(self.history)
@@ -177,6 +180,21 @@ CodeJade: "Arquivo criado. Quer que eu execute?"
             tool_data = self.process_tool_call(response)
             
             if tool_data:
+                # Loop detection
+                current_call_signature = f"{tool_data['tool']}:{json.dumps(tool_data['args'], sort_keys=True)}"
+                if last_tool_call == current_call_signature:
+                    consecutive_tool_failures += 1
+                    if consecutive_tool_failures >= 2:
+                        error_msg = f"TOOL_ERROR: Detected repetitive loop with same arguments. Stopping tool execution."
+                        self.history.append({"role": "system", "content": error_msg})
+                        print(f"🛑 Loop detectado: {error_msg}")
+                        # Force model to explain
+                        continue
+                else:
+                    consecutive_tool_failures = 0
+                
+                last_tool_call = current_call_signature
+
                 # É uma ferramenta -> Executa
                 tool_result = self.run_tool(tool_data)
                 
@@ -192,6 +210,13 @@ CodeJade: "Arquivo criado. Quer que eu execute?"
                 self.history.append({"role": "assistant", "content": final_response})
                 break
         
+        # Se saiu do loop sem resposta final (max_turns reached)
+        if not final_response and turn >= max_turns:
+             print("⚠️ Max turns reached without final response. Forcing summary.")
+             self.history.append({"role": "system", "content": "SYSTEM: Max tool turns reached. Please summarize what was done and what failed."})
+             final_response = self._chat(self.history)
+             self.history.append({"role": "assistant", "content": final_response})
+
         # Gerencia a memória ao fim do ciclo
         self._manage_memory()