Add file upload RAG: main.py + index.html

app/main.py

@@ -3,13 +3,14 @@
 import re
 from contextlib import asynccontextmanager
 from pathlib import Path
+from typing import List
 
-from fastapi import FastAPI
+from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import FileResponse
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 
-from app.rag import load_corpus, retrieve
+from app.rag import load_corpus, retrieve, add_documents, list_documents, delete_document
 from app.llm import build_system_prompt, chat, analyze_session
 
 
@@ -25,6 +26,8 @@ app = FastAPI(title="AIM Learning Companion", lifespan=lifespan)
 STATIC_DIR = Path(__file__).parent.parent / "static"
 app.mount("/static", StaticFiles(directory=str(STATIC_DIR)), name="static")
 
+ALLOWED_EXTENSIONS = {".txt", ".pdf", ".pptx", ".ppt", ".zip"}
+
 
 class ChatRequest(BaseModel):
     message: str
@@ -84,6 +87,39 @@ async def api_chat(req: ChatRequest):
     return ChatResponse(reply=reply, phase=detected_phase)
 
 
+@app.post("/api/upload")
+async def api_upload(files: List[UploadFile] = File(...)):
+    """Upload one or more files (PDF, PPTX, TXT, ZIP) to the RAG corpus."""
+    file_data = []
+    skipped = []
+
+    for f in files:
+        ext = Path(f.filename).suffix.lower() if f.filename else ""
+        if ext not in ALLOWED_EXTENSIONS:
+            skipped.append({"filename": f.filename, "reason": f"Type non supporte: {ext}"})
+            continue
+        content = await f.read()
+        file_data.append((f.filename, content))
+
+    results = add_documents(file_data) if file_data else []
+    return {"results": results, "skipped": skipped}
+
+
+@app.get("/api/documents")
+async def api_documents():
+    """List all documents in the corpus."""
+    return {"documents": list_documents()}
+
+
+@app.delete("/api/documents/{filename}")
+async def api_delete_document(filename: str):
+    """Delete a document from the corpus."""
+    ok = delete_document(filename)
+    if ok:
+        return {"status": "ok"}
+    return {"status": "error", "message": "Fichier non trouve"}
+
+
 @app.post("/api/analyze", response_model=AnalysisResponse)
 async def api_analyze(req: AnalysisRequest):
     analysis = await analyze_session(req.history)

static/index.html

@@ -18,6 +18,18 @@
             <input type="text" id="topic-input" placeholder="Ex : L'intelligence artificielle en formation professionnelle">
         </div>
 
+        <div class="form-group">
+            <label>Documents de reference (optionnel)</label>
+            <div class="upload-zone" id="upload-zone">
+                <div class="upload-icon">+</div>
+                <div class="upload-text">Glisse tes fichiers ici ou clique pour selectionner</div>
+                <div class="upload-hint">PDF, PPTX, TXT ou ZIP — plusieurs fichiers possibles</div>
+                <input type="file" id="file-input" multiple accept=".pdf,.pptx,.ppt,.txt,.zip" hidden>
+            </div>
+            <div class="upload-list" id="upload-list"></div>
+            <div class="upload-status" id="upload-status"></div>
+        </div>
+
         <div class="form-group">
             <label>Mode</label>
             <div class="mode-selector">
@@ -41,6 +53,7 @@
             <div class="chat-header-left">
                 <span id="mode-badge" class="badge badge-mode"></span>
                 <span id="topic-badge" class="badge badge-topic"></span>
+                <span id="docs-badge" class="badge badge-docs" style="display:none"></span>
             </div>
             <div class="chat-header-right">
                 <button id="btn-end-session" class="btn-end">Terminer la session</button>