Spaces:
Sleeping
Sleeping
Commit
·
7cf24a1
1
Parent(s):
e2385d4
Upd imports
Browse files- api/app.py +3 -3
- api/chatbot.py +2 -2
- api/routes.py +2 -2
- models/llama.py +1 -1
api/app.py
CHANGED
|
@@ -2,9 +2,9 @@
|
|
| 2 |
import uvicorn
|
| 3 |
from fastapi import FastAPI
|
| 4 |
from fastapi.middleware.cors import CORSMiddleware
|
| 5 |
-
from
|
| 6 |
-
from
|
| 7 |
-
from
|
| 8 |
|
| 9 |
# ✅ Validate environment
|
| 10 |
validate_environment()
|
|
|
|
| 2 |
import uvicorn
|
| 3 |
from fastapi import FastAPI
|
| 4 |
from fastapi.middleware.cors import CORSMiddleware
|
| 5 |
+
from config import setup_logging, check_system_resources, optimize_memory, CORS_ORIGINS, validate_environment
|
| 6 |
+
from database import db_manager
|
| 7 |
+
from routes import router
|
| 8 |
|
| 9 |
# ✅ Validate environment
|
| 10 |
validate_environment()
|
api/chatbot.py
CHANGED
|
@@ -3,8 +3,8 @@ import re
|
|
| 3 |
import logging
|
| 4 |
from typing import Dict
|
| 5 |
from google import genai
|
| 6 |
-
from
|
| 7 |
-
from
|
| 8 |
from memory import MemoryManager
|
| 9 |
from utils import translate_query, process_medical_image
|
| 10 |
from search import search_web
|
|
|
|
| 3 |
import logging
|
| 4 |
from typing import Dict
|
| 5 |
from google import genai
|
| 6 |
+
from config import gemini_flash_api_key
|
| 7 |
+
from retrieval import retrieval_engine
|
| 8 |
from memory import MemoryManager
|
| 9 |
from utils import translate_query, process_medical_image
|
| 10 |
from search import search_web
|
api/routes.py
CHANGED
|
@@ -3,8 +3,8 @@ import time
|
|
| 3 |
import logging
|
| 4 |
from fastapi import APIRouter, Request
|
| 5 |
from fastapi.responses import JSONResponse
|
| 6 |
-
from
|
| 7 |
-
from
|
| 8 |
from utils import process_medical_image
|
| 9 |
|
| 10 |
logger = logging.getLogger("medical-chatbot")
|
|
|
|
| 3 |
import logging
|
| 4 |
from fastapi import APIRouter, Request
|
| 5 |
from fastapi.responses import JSONResponse
|
| 6 |
+
from chatbot import RAGMedicalChatbot
|
| 7 |
+
from retrieval import retrieval_engine
|
| 8 |
from utils import process_medical_image
|
| 9 |
|
| 10 |
logger = logging.getLogger("medical-chatbot")
|
models/llama.py
CHANGED
|
@@ -42,7 +42,7 @@ Keywords:"""
|
|
| 42 |
"""Use Llama to summarize documents and return summary with URL mapping"""
|
| 43 |
try:
|
| 44 |
# Import summarizer here to avoid circular imports
|
| 45 |
-
from
|
| 46 |
|
| 47 |
# Use the summarizer for document summarization
|
| 48 |
combined_summary, url_mapping = summarizer.summarize_documents(documents, user_query)
|
|
|
|
| 42 |
"""Use Llama to summarize documents and return summary with URL mapping"""
|
| 43 |
try:
|
| 44 |
# Import summarizer here to avoid circular imports
|
| 45 |
+
from summarizer import summarizer
|
| 46 |
|
| 47 |
# Use the summarizer for document summarization
|
| 48 |
combined_summary, url_mapping = summarizer.summarize_documents(documents, user_query)
|