Update backend/main.py

backend/main.py

@@ -1,69 +1,69 @@
-from fastapi import FastAPI, UploadFile, File, Form
-from pydantic import BaseModel
-import shutil
-import os
-
-from backend.models import summarizer, image_captioning, qa_pipeline, translator, code_generator
-from backend.utils import extract_text_from_pdf, extract_text_from_document
-
-app = FastAPI()
-
-UPLOAD_DIR = "uploads"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
-
-class QARequest(BaseModel):
-    question: str
-    context: str
-
-@app.get("/")
-async def home():
-    return {"message": "AI Web App is Running!"}
-
-# 1️⃣ استخراج النصوص من المستندات
-@app.post("/extract_text/")
-async def extract_text(file: UploadFile = File(...)):
-    file_path = f"{UPLOAD_DIR}/{file.filename}"
-    with open(file_path, "wb") as f:
-        shutil.copyfileobj(file.file, f)
-
-    text = extract_text_from_document(file_path)
-    os.remove(file_path)
-    return {"text": text}
-
-# 2️⃣ تلخيص المستندات
-@app.post("/summarize/")
-async def summarize_text(text: str = Form(...)):
-    summary = summarizer(text)
-    return {"summary": summary[0]['summary_text']}
-
-# 3️⃣ تحليل الصور وتوليد وصف لها
-@app.post("/image_caption/")
-async def caption_image(file: UploadFile = File(...)):
-    file_path = f"{UPLOAD_DIR}/{file.filename}"
-    with open(file_path, "wb") as f:
-        shutil.copyfileobj(file.file, f)
-
-    with open(file_path, "rb") as img:
-        image_bytes = img.read()
-
-    caption = image_captioning(image_bytes)
-    os.remove(file_path)
-    return {"caption": caption[0]['generated_text']}
-
-# 4️⃣ الإجابة على الأسئلة من النصوص
-@app.post("/qa/")
-async def answer_question(request: QARequest):
-    answer = qa_pipeline(request.question, request.context)
-    return {"answer": answer["answer"]}
-
-# 5️⃣ ترجمة المستندات
-@app.post("/translate/")
-async def translate_text(text: str = Form(...)):
-    translation = translator(text)
-    return {"translation": translation[0]['translation_text']}
-
-# 6️⃣ توليد أكواد التصور من أوامر نصية
-@app.post("/generate_code/")
-async def generate_code(prompt: str = Form(...)):
-    code = code_generator(prompt)
-    return {"code": code}
+from fastapi import FastAPI, UploadFile, File, Form
+from pydantic import BaseModel
+import shutil
+import os
+
+from backend.models import summarizer, image_captioning, qa_pipeline, translator, code_generator
+from backend.utils import extract_text_from_pdf, extract_text_from_document
+
+app = FastAPI()
+
+UPLOAD_DIR = "/tmp/uploads"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
+class QARequest(BaseModel):
+    question: str
+    context: str
+
+@app.get("/")
+async def home():
+    return {"message": "AI Web App is Running!"}
+
+# 1️⃣ استخراج النصوص من المستندات
+@app.post("/extract_text/")
+async def extract_text(file: UploadFile = File(...)):
+    file_path = f"{UPLOAD_DIR}/{file.filename}"
+    with open(file_path, "wb") as f:
+        shutil.copyfileobj(file.file, f)
+
+    text = extract_text_from_document(file_path)
+    os.remove(file_path)
+    return {"text": text}
+
+# 2️⃣ تلخيص المستندات
+@app.post("/summarize/")
+async def summarize_text(text: str = Form(...)):
+    summary = summarizer(text)
+    return {"summary": summary[0]['summary_text']}
+
+# 3️⃣ تحليل الصور وتوليد وصف لها
+@app.post("/image_caption/")
+async def caption_image(file: UploadFile = File(...)):
+    file_path = f"{UPLOAD_DIR}/{file.filename}"
+    with open(file_path, "wb") as f:
+        shutil.copyfileobj(file.file, f)
+
+    with open(file_path, "rb") as img:
+        image_bytes = img.read()
+
+    caption = image_captioning(image_bytes)
+    os.remove(file_path)
+    return {"caption": caption[0]['generated_text']}
+
+# 4️⃣ الإجابة على الأسئلة من النصوص
+@app.post("/qa/")
+async def answer_question(request: QARequest):
+    answer = qa_pipeline(request.question, request.context)
+    return {"answer": answer["answer"]}
+
+# 5️⃣ ترجمة المستندات
+@app.post("/translate/")
+async def translate_text(text: str = Form(...)):
+    translation = translator(text)
+    return {"translation": translation[0]['translation_text']}
+
+# 6️⃣ توليد أكواد التصور من أوامر نصية
+@app.post("/generate_code/")
+async def generate_code(prompt: str = Form(...)):
+    code = code_generator(prompt)
+    return {"code": code}