hjsu

Dockerfile

@@ -2,4 +2,5 @@ FROM python:3.10-slim
 COPY . /app
 WORKDIR /app
 RUN pip install -r requirements.txt
-CMD ["python", "app.py"]
+RUN apt-get update && apt-get install -y ffmpeg
+CMD ["uvicorn", "fastapi_app:app", "--host", "0.0.0.0", "--port", "7860"]

audio_utils.py

@@ -0,0 +1,24 @@
+import numpy as np
+import noisereduce as nr
+import soundfile as sf
+from pydub import AudioSegment
+import io
+
+def preprocess_audio(file_bytes):
+    # Charger l'audio
+    audio = AudioSegment.from_file(io.BytesIO(file_bytes))
+    # Convertir en mono, 16kHz
+    audio = audio.set_channels(1).set_frame_rate(16000)
+    # Exporter en wav bytes
+    wav_io = io.BytesIO()
+    audio.export(wav_io, format='wav')
+    wav_io.seek(0)
+    # Charger en numpy
+    y, sr = sf.read(wav_io)
+    # Réduction de bruit
+    y_denoised = nr.reduce_noise(y=y, sr=sr)
+    # Sauvegarder en wav bytes
+    out_io = io.BytesIO()
+    sf.write(out_io, y_denoised, sr, format='WAV')
+    out_io.seek(0)
+    return out_io 

fastapi_app.py

@@ -2,6 +2,8 @@
 """
 Medical AI Assistant - FastAPI Only Version
 Simplified endpoints for backend integration with Swagger UI
+
+This file is Hugging Face Spaces compatible: the FastAPI app is exposed as 'app' at the module level.
 """
 
 from fastapi import FastAPI, HTTPException, File, UploadFile, BackgroundTasks
@@ -62,15 +64,15 @@ async def load_models():
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    """Application lifespan management"""
+    """Application lifespan management (robust for Hugging Face Spaces)"""
     try:
         await load_models()
-        yield
+        logger.info("✅ Models loaded in lifespan.")
     except Exception as e:
         logger.error(f"❌ Error during startup: {str(e)}", exc_info=True)
-        raise
-    finally:
-        logger.info("🔄 Shutting down...")
+        # Do not raise, just log. App will start but endpoints will return 503 if models are missing.
+    yield
+    logger.info("🔄 Shutting down...")
 
 # Custom OpenAPI schema
 def custom_openapi():
@@ -594,21 +596,20 @@ async def validation_exception_handler(request, exc):
         }
     )
 
-# ============================================================================
+# =========================================================================
 # STARTUP MESSAGE
-# ============================================================================
+# =========================================================================
 
-if __name__ == "__main__":
-    import uvicorn
-    
-    print("🩺 Starting Medical AI Assistant API...")
-    print("📚 Documentation available at: http://localhost:8000/docs")
-    print("🔄 Alternative docs at: http://localhost:8000/redoc")
-    
-    uvicorn.run(
-        app,
-        host="0.0.0.0",
-        port=8000,
-        log_level="info",
-        reload=False
-    )
+# The following block is removed for Hugging Face Spaces compatibility:
+# if __name__ == "__main__":
+#     import uvicorn
+#     print("🩺 Starting Medical AI Assistant API...")
+#     print("📚 Documentation available at: http://localhost:8000/docs")
+#     print("🔄 Alternative docs at: http://localhost:8000/redoc")
+#     uvicorn.run(
+#         app,
+#         host="0.0.0.0",
+#         port=8000,
+#         log_level="info",
+#         reload=False
+#     )

requirements.txt

@@ -36,4 +36,7 @@ pytest==7.4.3
 pytest-asyncio==0.21.1
 
 # Optional: For production deployment
-gunicorn==21.2.0
+gunicorn==21.2.0
+
+noisereduce==3.0.1
+pydub==0.25.1