Update app.py

app.py

@@ -1,40 +1,92 @@
 # app.py
 
-
 from fastapi import FastAPI, HTTPException, UploadFile, File, Form
 from typing import List  # Import List from typing
 from schemas import LoadModelRequest, ChatRequest
-
-    logger.error(f"Startup failed: {str(e)}")
-       raise
-
-
-
-    @app.post("/load-model/")
-    async def load_model(request: LoadModelRequest):
-        return model_service.load_model(request.model_name, request.temperature)
-            logger.error(f"Error listing models: {str(e)}")
-            raise HTTPException(status_code=500, detail="Failed to list available models")
-    
-    
-    @app.get("/list-models/")
-    async def list_models():
-        """List all models (chatbots) available in Google Drive."""
-            logger.error(f"Error listing models: {str(e)}")
-            raise HTTPException(status_code=500, detail=f"Failed to list models: {str(e)}")
-    
-    
-    
-    
-    @app.post("/update-chatbot/")
-    async def update_chatbot(
-        folder_name: str = Form(...),
-    ):
-        """Update an existing chatbot by appending new data (PDF files)."""
-        return chatbot_service.update_chatbot(folder_name, pdf_files)
-
-
-
+from drive_service import DriveService
+from chatbot_service import ChatbotService
+from model_service import ModelService
+from config import BASE_MODEL_PATH, CUSTOM_CHATBOTS_DIR
+from config import GOOGLE_DRIVE_FOLDER_ID  # Import GOOGLE_DRIVE_FOLDER_ID
+
+import os
+import logging
+
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+# Initialize services
+drive_service = DriveService()
+chatbot_service = ChatbotService()
+model_service = ModelService()
+
+@app.on_event("startup")
+async def startup_event():
+    """Initialize necessary components on startup."""
+    try:
+        logger.info("Application started successfully")
+    except Exception as e:
+        logger.error(f"Startup failed: {str(e)}")
+        raise
+
+@app.post("/load-model/")
+async def load_model(request: LoadModelRequest):
+    return model_service.load_model(request.model_name, request.temperature)
+
+@app.post("/chat/{model_name}")
+async def chat_with_model(model_name: str, request: ChatRequest):
+    return model_service.chat_with_model(model_name, request.question)
+
+@app.post("/create-chatbot/")
+async def create_chatbot(
+    folder_name: str = Form(...),
+    pdf_files: List[UploadFile] = File(...)  # Now List is imported
+):
+    return chatbot_service.create_chatbot(folder_name, pdf_files)
+
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy"}
+
+@app.get("/")
+async def root():
+    return {"message": "API is running"}
+
+@app.get("/available-models")
+async def list_available_models():
+    try:
+        models = [name for name in os.listdir(BASE_MODEL_PATH) if os.path.isdir(os.path.join(BASE_MODEL_PATH, name))]
+        return {
+            "status": "success",
+            "models": models
+        }
+    except Exception as e:
+        logger.error(f"Error listing models: {str(e)}")
+        raise HTTPException(status_code=500, detail="Failed to list available models")
+
+@app.get("/list-models/")
+async def list_models():
+    """List all models (chatbots) available in Google Drive."""
+    try:
+        # List all folders in the Google Drive folder
+        folders = drive_service.list_files_in_folder(GOOGLE_DRIVE_FOLDER_ID)
+        models = [folder['name'] for folder in folders if folder['mimeType'] == 'application/vnd.google-apps.folder']
+        return {
+            "status": "success",
+            "models": models
+        }
+    except Exception as e:
+        logger.error(f"Error listing models: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to list models: {str(e)}")
+
+@app.post("/update-chatbot/")
+async def update_chatbot(
+    folder_name: str = Form(...),
+    pdf_files: List[UploadFile] = File(...)
+):
+    """Update an existing chatbot by appending new data (PDF files)."""
+    return chatbot_service.update_chatbot(folder_name, pdf_files)
 
 # import os
 # import io