brendon-ai commited on
Commit
766487e
·
verified ·
1 Parent(s): 1aba25d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -3,6 +3,7 @@ from fastapi import FastAPI, HTTPException
3
  from pydantic import BaseModel
4
  from typing import Optional
5
  import uvicorn
 
6
  from src.RAGSample import setup_retriever, setup_rag_chain, RAGApplication
7
  import os
8
  from dotenv import load_dotenv
@@ -154,5 +155,19 @@ async def get_models():
154
  "vector_database": "ChromaDB (local)"
155
  }
156
 
 
 
 
 
157
  if __name__ == "__main__":
158
- uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")
 
 
 
 
 
 
 
 
 
 
 
3
  from pydantic import BaseModel
4
  from typing import Optional
5
  import uvicorn
6
+ import logging
7
  from src.RAGSample import setup_retriever, setup_rag_chain, RAGApplication
8
  import os
9
  from dotenv import load_dotenv
 
155
  "vector_database": "ChromaDB (local)"
156
  }
157
 
158
+
159
+ logging.basicConfig(level=logging.INFO)
160
+ logger = logging.getLogger(__name__)
161
+
162
  if __name__ == "__main__":
163
+ try:
164
+ logger.info("Starting application...")
165
+ # Add any initialization code here with try/except blocks
166
+
167
+ port = int(os.getenv("PORT", 7860))
168
+ logger.info(f"Starting server on port {port}")
169
+
170
+ uvicorn.run(app, host="0.0.0.0", port=port, log_level="info")
171
+ except Exception as e:
172
+ logger.error(f"Failed to start application: {e}")
173
+ raise