Spaces:
Sleeping
Sleeping
File size: 2,437 Bytes
4254fda e2c1ba7 717f924 4254fda 717f924 e2c1ba7 717f924 e2c1ba7 717f924 e2c1ba7 021d4f9 717f924 021d4f9 717f924 4254fda 717f924 4254fda 021d4f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import gradio as gr
from rag_pipeline import RAGPipeline
from utils import process_documents
import asyncio
import time
rag = RAGPipeline()
def log_message(msg, logs):
logs.append(msg)
return logs
def upload_and_index(files, logs):
logs = log_message("[RAG] بدء معالجة الملفات...", logs)
all_chunks = []
for file in files:
logs = log_message(f"[RAG] معالجة الملف: {file.name}", logs)
chunks = process_documents(file.name)
all_chunks.extend(chunks)
logs = log_message(f"[RAG] تم استخراج {len(chunks)} مقطع من {file.name}", logs)
logs = log_message(f"[RAG] بناء الفهرس لـ {len(all_chunks)} مقطع...", logs)
start = time.time()
rag.build_index(all_chunks, logs)
duration = time.time() - start
logs = log_message(f"[RAG] تم بناء الفهرس في {duration:.2f} ثانية.", logs)
return logs, True # True لتمكين مربع السؤال
def answer_question(question, logs):
logs = log_message(f"[RAG] استلام السؤال: {question}", logs)
start = time.time()
answer, sources = rag.answer(question)
duration = time.time() - start
logs = log_message(f"[RAG] تم الإجابة في {duration:.2f} ثانية.", logs)
logs = log_message(f"[RAG] المصادر: {sources}", logs)
return answer, logs
with gr.Blocks() as demo:
logs = gr.State([])
gr.Markdown("# نظام استرجاع المعرفة من الملفات (RAG)")
with gr.Row():
files_input = gr.File(file_types=['.pdf', '.docx', '.txt'], file_count="multiple", label="رفع الملفات")
upload_btn = gr.Button("رفع وبناء الفهرس")
logs_output = gr.Textbox(label="سجل العمليات", lines=10, interactive=False)
question_input = gr.Textbox(label="اكتب سؤالك هنا", visible=False)
ask_btn = gr.Button("إرسال السؤال", visible=False)
answer_output = gr.Textbox(label="الإجابة", lines=5)
upload_btn.click(upload_and_index, inputs=[files_input, logs], outputs=[logs_output, question_input])
upload_btn.click(lambda: True, None, question_input) # اظهار مربع السؤال
upload_btn.click(lambda: True, None, ask_btn) # اظهار زر السؤال
ask_btn.click(answer_question, inputs=[question_input, logs], outputs=[answer_output, logs_output])
demo.launch()
|