Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py - FastAPI application
 import os
 import sys
 import json
@@ -11,6 +10,14 @@ import torch
 from datetime import datetime
 from pydantic import BaseModel
 
+# Add src directory to Python path
+current_dir = os.path.dirname(os.path.abspath(__file__))
+src_path = os.path.abspath(os.path.join(current_dir, "src"))
+sys.path.insert(0, src_path)
+
+# Now import TxAgent after adding to path
+from txagent.txagent import TxAgent
+
 # Configuration
 persistent_dir = "/data/hf_cache"
 model_cache_dir = os.path.join(persistent_dir, "txagent_models")
@@ -28,11 +35,6 @@ os.makedirs(report_dir, exist_ok=True)
 os.environ["HF_HOME"] = model_cache_dir
 os.environ["TRANSFORMERS_CACHE"] = model_cache_dir
 
-# Set up Python path
-current_dir = os.path.dirname(os.path.abspath(__file__))
-src_path = os.path.abspath(os.path.join(current_dir, "src"))
-sys.path.insert(0, src_path)
-
 # Request models
 class ChatRequest(BaseModel):
     message: str
@@ -70,7 +72,9 @@ async def startup_event():
     global agent
     try:
         agent = init_agent()
+        logger.info("TxAgent initialized successfully")
     except Exception as e:
+        logger.error(f"Failed to initialize agent: {str(e)}")
         raise RuntimeError(f"Failed to initialize agent: {str(e)}")
 
 def init_agent():