Spaces:

yousbek
/

questionanswering

Sleeping

App Files Files Community

yousbek commited on Mar 21, 2025

Commit

c536a32

verified ·

1 Parent(s): 11c0208

Upload 8 files

Browse files

Files changed (8) hide show

Dockerfile +37 -0
app/__init__.py +0 -0
app/main.py +53 -0
app/utils.py +5 -0
requirements.txt +0 -0
static/index.html +37 -0
static/script.js +35 -0
static/styles.css +30 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,37 @@

+# Use Python 3.9.11
+FROM python:3.9.11
+# Set the working directory
+WORKDIR /code
+# Copy the requirements file
+COPY ./requirements.txt /code/requirements.txt
+# Install dependencies
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Add a non-root user
+RUN useradd -m user
+# Set environment variables
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+# Create the cache directory and set permissions
+RUN mkdir -p /home/user/.cache/huggingface && chown -R user:user /home/user/.cache
+# Switch to the non-root user
+USER user
+# Set the working directory for the user
+WORKDIR $HOME/app
+# Copy the content into /home/user/app with the correct permissions
+COPY --chown=user:user . $HOME/app
+# Expose the port the app runs on
+EXPOSE 7860
+# Command to run the application
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/__init__.py ADDED Viewed

File without changes

app/main.py ADDED Viewed

	@@ -0,0 +1,53 @@

+from fastapi import FastAPI, File, UploadFile, Form ,HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse, FileResponse
+from fastapi.staticfiles import StaticFiles
+import pytesseract
+from PIL import Image
+import PyPDF2
+from transformers import pipeline
+import os
+app = FastAPI()
+# Autoriser les requêtes CORS (pour l'interface web)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Autoriser toutes les origines
+    allow_methods=["*"],  # Autoriser toutes les méthodes
+    allow_headers=["*"],  # Autoriser tous les en-têtes
+)
+# Charger le modèle de question-réponse
+qa_pipeline = pipeline("question-answering", model="bert-large-uncased-whole-word-masking-finetuned-squad")
+def extract_text_from_image(image):
+    return pytesseract.image_to_string(image)
+def extract_text_from_pdf(file):
+    reader = PyPDF2.PdfReader(file)
+    text = ""
+    for page in reader.pages:
+        text += page.extract_text()
+    return text
+# Endpoint pour répondre à des questions
+@app.post("/ask")
+async def ask_question(file: UploadFile = File(...), question: str = Form(...)):
+    # Extraire le texte en fonction du type de fichier
+    if file.filename.endswith(".pdf"):
+        text = extract_text_from_pdf(file.file)
+    elif file.filename.endswith((".png", ".jpg", ".jpeg")):
+        image = Image.open(file.file)
+        text = extract_text_from_image(image)
+    else:
+        return {"answer": "Format de fichier non supporté."}
+    # Répondre à la question
+    result = qa_pipeline(question=question, context=text)
+    return {"answer": result["answer"]}
+# Démarrer l'application
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

app/utils.py ADDED Viewed

	@@ -0,0 +1,5 @@

+import pytesseract
+from PIL import Image
+import PyPDF2
+from docx import Document
+import pandas as pd

requirements.txt ADDED Viewed

Binary file (1.8 kB). View file

static/index.html ADDED Viewed

	@@ -0,0 +1,37 @@

+<!DOCTYPE html>
+<html lang="fr">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Service de Question-Réponse Intelligent</title>
+    <link rel="stylesheet" href="styles.css">
+</head>
+<body>
+    <div class="container">
+        <h1>Service de Question-Réponse Intelligent</h1>
+        <!-- Section pour téléverser un fichier -->
+        <div class="upload-section">
+            <label for="file-upload">Téléverser un document ou une image :</label>
+            <input type="file" id="file-upload" accept=".pdf,.png,.jpg,.jpeg">
+        </div>
+        <!-- Section pour poser une question -->
+        <div class="question-section">
+            <label for="question">Posez votre question :</label>
+            <textarea id="question" rows="3" placeholder="Exemple : Quel est le sujet de ce document ?"></textarea>
+        </div>
+        <!-- Bouton pour soumettre -->
+        <button id="submit-btn">Soumettre</button>
+        <!-- Section pour afficher les résultats -->
+        <div class="result-section">
+            <h2>Réponse :</h2>
+            <p id="answer"></p>
+        </div>
+    </div>
+    <script src="script.js"></script>
+</body>
+</html>

static/script.js ADDED Viewed

	@@ -0,0 +1,35 @@

+document.getElementById("submit-btn").addEventListener("click", async () => {
+    const fileInput = document.getElementById("file-upload");
+    const questionInput = document.getElementById("question");
+    const answerOutput = document.getElementById("answer");
+    const file = fileInput.files[0];
+    const question = questionInput.value;
+    if (!file || !question) {
+        alert("Veuillez téléverser un fichier et poser une question.");
+        return;
+    }
+    // Créer un objet FormData pour envoyer le fichier et la question
+    const formData = new FormData();
+    formData.append("file", file);
+    formData.append("question", question);
+    try {
+        const response = await fetch("http://localhost:7860/ask", {
+            method: "POST",
+            body: formData,
+        });
+        if (!response.ok) {
+            throw new Error("Erreur lors de la requête.");
+        }
+        const data = await response.json();
+        answerOutput.textContent = data.answer || "Aucune réponse trouvée.";
+    } catch (error) {
+        console.error(error);
+        answerOutput.textContent = "Une erreur s'est produite. Veuillez réessayer.";
+    }
+});

static/styles.css ADDED Viewed

	@@ -0,0 +1,30 @@

+body {
+    font-family: Arial, sans-serif;
+    margin: 20px;
+}
+.container {
+    max-width: 600px;
+    margin: 0 auto;
+}
+h1 {
+    text-align: center;
+}
+.upload-section, .question-section, .result-section {
+    margin-bottom: 20px;
+}
+textarea, input[type="file"], button {
+    width: 100%;
+    padding: 10px;
+    margin-top: 10px;
+}
+.result-section {
+    padding: 20px;
+    background-color: #f9f9f9;
+    border: 1px solid #ddd;
+    border-radius: 5px;
+}