Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|