removed all logs

Author: codinglabsong

api/llm/agent.py

@@ -1,5 +1,4 @@
 import atexit
-import logging
 import os
 import re
 from datetime import datetime
@@ -16,10 +15,6 @@
 
 load_dotenv()
 
-# Configure logging
-logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
-logger = logging.getLogger(__name__)
-
 # Global agent instance
 _agent_executor = None
 _checkpointer = None
@@ -121,18 +116,11 @@ def chat_with_agent(message: str, user_id: str = "default", selected_images: Opt
             agent_response = last_message["content"]
 
         # Check if any tools were used (image generation)
-        logger.info(f"Response keys: {response.keys()}")
-        logger.info(f"Has intermediate_steps: {'intermediate_steps' in response}")
         if "intermediate_steps" in response and response["intermediate_steps"]:
-            logger.info(f"Intermediate steps: {response['intermediate_steps']}")
             for step in response["intermediate_steps"]:
-                logger.info(f"Step: {step}")
-                logger.info(f"Step[0] type: {type(step[0])}")
-                logger.info(f"Step[0] str: {str(step[0])}")
                 if len(step) >= 2 and "generate_image" in str(step[0]):
                     # Extract image data from the tool result
                     tool_result = step[1]
-                    logger.info(f"Tool result: {tool_result}")
 
                     # Try multiple patterns to find the image ID
                     image_id = None
@@ -189,7 +177,7 @@ def chat_with_agent(message: str, user_id: str = "default", selected_images: Opt
                                 }
 
                             except Exception as e:
-                                logger.error(f"Error getting S3 metadata: {e}")
+                                print(f"Error getting S3 metadata: {e}")
                                 # Don't return image data if we can't get a valid URL
                                 generated_image_data = None
                                 # Add error message to agent response

api/server/main.py

@@ -1,15 +1,10 @@
-import logging
 from typing import Dict, List, Optional
 
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 
 from llm.agent import chat_with_agent
 
-# Configure logging
-logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
-logger = logging.getLogger(__name__)
-
 app = FastAPI(
     title="AI Image Editor API",
     description="API for AI-powered image editing assistant",
@@ -40,26 +35,14 @@ class ChatResponse(BaseModel):
 
 @app.get("/")
 async def root():
-    logger.info("Root endpoint accessed")
     return {"message": "AI Image Editor API is running!"}
 
 
 @app.get("/health")
 async def health_check():
-    logger.info("Health check endpoint accessed")
     return {"status": "healthy", "service": "ai-image-editor-api"}
 
 
-@app.get("/logs")
-async def get_logs():
-    """Get recent logs for debugging (development only)"""
-    logger.info("Logs endpoint accessed")
-    return {
-        "message": "Check Hugging Face Space logs for detailed information",
-        "logs_url": "https://huggingface.co/spaces/YOUR_USERNAME/YOUR_SPACE_NAME/logs",
-    }
-
-
 @app.post("/chat", response_model=ChatResponse)
 async def chat_endpoint(request: ChatRequest):
     """
@@ -72,8 +55,6 @@ async def chat_endpoint(request: ChatRequest):
         ChatResponse with AI response, status, and optional generated image metadata.
     """
     try:
-        logger.info(f"Chat request received - user_id: {request.user_id}, message: {request.message[:100]}...")
-
         # Use the LLM agent to get a response
         user_id = request.user_id or "default"
         response, generated_image_data = chat_with_agent(
@@ -82,19 +63,15 @@ async def chat_endpoint(request: ChatRequest):
             selected_images=request.selected_images,
         )
 
-        logger.info(f"Agent response received - has generated image: {generated_image_data is not None}")
-
         # Create response with optional generated image
         chat_response = ChatResponse(response=response, status="success")
 
         if generated_image_data:
             chat_response.generated_image = GeneratedImage(**generated_image_data)
-            logger.info(f"Generated image data: {generated_image_data}")
 
         return chat_response
 
     except Exception as e:
-        logger.error(f"Error processing chat request: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Error processing request: {str(e)}")