Create app.py

app.py

@@ -0,0 +1,56 @@
+from fastapi import FastAPI, Depends, HTTPException, Security
+from fastapi.security.api_key import APIKeyHeader
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from transformers import pipeline
+import os
+
+# سحب الباسوورد من الإعدادات
+SECRET_API_KEY = os.environ.get("API_KEY")
+api_key_header = APIKeyHeader(name="X-API-Key", auto_error=False)
+
+app = FastAPI(title="Dentor NLP API")
+
+# السماح للموقع بتبادل البيانات
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+def get_api_key(api_key_header: str = Security(api_key_header)):
+    if api_key_header == SECRET_API_KEY:
+        return api_key_header
+    raise HTTPException(status_code=403, detail="Access Denied: Invalid API Key")
+
+# تحميل الموديلات (من الفولدرات اللي لسه رافعينها)
+print("Loading BioBART Summarizer...")
+summarizer = pipeline("summarization", model="./biobart", tokenizer="./biobart")
+
+print("Loading Helsinki Translator...")
+translator = pipeline("translation", model="./helsinki", tokenizer="./helsinki")
+
+print("All Models Loaded!")
+
+class TextRequest(BaseModel):
+    text: str
+
+# مسار التلخيص
+@app.post("/summarize")
+async def summarize_text(request: TextRequest, api_key: str = Depends(get_api_key)):
+    try:
+        summary = summarizer(request.text, max_length=150, min_length=30, do_sample=False)
+        return {"status": "success", "summary": summary[0]['summary_text']}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# مسار الترجمة
+@app.post("/translate")
+async def translate_text(request: TextRequest, api_key: str = Depends(get_api_key)):
+    try:
+        translation = translator(request.text)
+        return {"status": "success", "translation": translation[0]['translation_text']}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))