Update backend/models.py

backend/models.py

@@ -1,6 +1,8 @@
 import os
 import requests
 
+from pdfminer.high_level import extract_text
+from docx import Document
 # تحميل متغيرات البيئة
 HUGGING_FACE_API_KEY = os.environ.get("HUGGING_FACE_API_KEY")
 
@@ -18,8 +20,11 @@ def summarizer(text):
     # ✅ طباعة حالة الاستجابة لمعرفة هل هناك خطأ؟
     print("API Response Status Code:", response.status_code)
     
-    # ✅ طباعة محتوى الاستجابة بالكامل
-    print("API Response:", response.json())
+    try:
+        response_json = response.json()  # تحويل الاستجابة إلى JSON
+        print("API Response:", response_json)
+    except Exception as e:
+        print("Error parsing response JSON:", e)
 
     return response.json()
 
@@ -46,7 +51,7 @@ def translator(text, source_lang, target_lang):
     # تحقق من أن اللغات مدعومة
     supported_langs = ['en', 'fr', 'es', 'ar', 'zh']
     if source_lang not in supported_langs or target_lang not in supported_langs:
-        return {"error": "اللغة غير مدعومة. اللغات المدعومة: en, fr, es, ar, zh"}
+        return {"error": "there is not support , supported languages: en, fr, es, ar, zh"}
 
     # توليد اسم النموذج بناءً على اللغات
     model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
@@ -66,4 +71,24 @@ def translator(text, source_lang, target_lang):
 
 # دالة توليد كود التصور البياني
 def code_generator(prompt):
-    return f"import matplotlib.pyplot as plt\n# {prompt}"
+    url = "https://api-inference.huggingface.co/models/Salesforce/codegen-350M-multi"
+    headers = {"Authorization": f"Bearer {HUGGING_FACE_API_KEY}"}
+    response = requests.post(url, headers=headers, json={"inputs": prompt})
+    
+    try:
+        return response.json()
+    except Exception as e:
+        return {"error": str(e)}
+
+
+def extract_text_from_document(path):
+    if path.endswith(".pdf"):
+        return extract_text(path)
+    elif path.endswith(".docx"):
+        doc = Document(path)
+        return "\n".join([p.text for p in doc.paragraphs])
+    elif path.endswith(".txt"):
+        with open(path, encoding="utf-8") as f:
+            return f.read()
+    else:
+        return "Unsupported file format."