almmyz321 commited on
Commit
e8d23f1
·
verified ·
1 Parent(s): 81cd3c3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import fitz # PyMuPDF
2
+ import gradio as gr
3
+ from transformers import pipeline
4
+
5
+ # تحميل النموذج العربي
6
+ qa = pipeline("question-answering", model="aubmindlab/bert-base-arabertv2")
7
+
8
+ def answer_pdf(file, question):
9
+ doc = fitz.open(stream=file.read(), filetype="pdf")
10
+ text = ""
11
+ for page in doc:
12
+ text += page.get_text()
13
+
14
+ if not question.strip():
15
+ return "يرجى كتابة سؤال لبدء التحليل."
16
+
17
+ result = qa(question=question, context=text[:4000]) # قص النص لتفادي الطول الزائد
18
+ return result["answer"]
19
+
20
+ gr.Interface(
21
+ fn=answer_pdf,
22
+ inputs=[
23
+ gr.File(label="ملف PDF"),
24
+ gr.Textbox(label="ما سؤالك؟", placeholder="مثال: ما هي الأدوات المذكورة؟")
25
+ ],
26
+ outputs=gr.Textbox(label="الإجابة"),
27
+ title="تحليل ملفات PDF بالعربية",
28
+ description="ارفع ملف PDF واسأل سؤالًا وسيتم الإجابة باستخدام نموذج Arabic BERT"
29
+ ).launch()