Upd imports

api/app.py

@@ -2,9 +2,9 @@
 import uvicorn
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from config import setup_logging, check_system_resources, optimize_memory, CORS_ORIGINS, validate_environment
-from database import db_manager
-from routes import router
+from .config import setup_logging, check_system_resources, optimize_memory, CORS_ORIGINS, validate_environment
+from .database import db_manager
+from .routes import router
 
 # ✅ Validate environment
 validate_environment()

api/chatbot.py

@@ -3,8 +3,8 @@ import re
 import logging
 from typing import Dict
 from google import genai
-from config import gemini_flash_api_key
-from retrieval import retrieval_engine
+from .config import gemini_flash_api_key
+from .retrieval import retrieval_engine
 from memory import MemoryManager
 from utils import translate_query, process_medical_image
 from search import search_web

api/database.py

@@ -4,7 +4,7 @@ import numpy as np
 import gridfs
 from pymongo import MongoClient
 from sentence_transformers import SentenceTransformer
-from api.config import mongo_uri, index_uri, MODEL_CACHE_DIR, EMBEDDING_MODEL_DEVICE
+from .config import mongo_uri, index_uri, MODEL_CACHE_DIR, EMBEDDING_MODEL_DEVICE
 import logging
 
 logger = logging.getLogger("medical-chatbot")

api/retrieval.py

@@ -1,7 +1,7 @@
 # api/retrieval.py
 import numpy as np
 import logging
-from api.database import db_manager
+from .database import db_manager
 
 logger = logging.getLogger("medical-chatbot")
 

api/routes.py

@@ -3,8 +3,8 @@ import time
 import logging
 from fastapi import APIRouter, Request
 from fastapi.responses import JSONResponse
-from chatbot import RAGMedicalChatbot
-from retrieval import retrieval_engine
+from .chatbot import RAGMedicalChatbot
+from .retrieval import retrieval_engine
 from utils import process_medical_image
 
 logger = logging.getLogger("medical-chatbot")

models/summarizer.py

@@ -1,7 +1,7 @@
 import re
 import logging
 from typing import List, Dict, Tuple
-from llama import NVIDIALLamaClient
+from .llama import NVIDIALLamaClient
 
 logger = logging.getLogger(__name__)
 

utils/__init__.py

@@ -1,3 +1,4 @@
 # Utils package
 from .translation import translate_query
 from .vlm import process_medical_image
+from .diagnosis import retrieve_diagnosis_from_symptoms