Spaces:
Runtime error
Runtime error
Delete FINAL_CODE.py
Browse files- FINAL_CODE.py +0 -191
FINAL_CODE.py
DELETED
@@ -1,191 +0,0 @@
|
|
1 |
-
import pandas as pd
|
2 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
-
from sentence_transformers import SentenceTransformer
|
4 |
-
from difflib import get_close_matches
|
5 |
-
import requests
|
6 |
-
import io
|
7 |
-
|
8 |
-
# GitHub Excel file URL
|
9 |
-
GITHUB_EXCEL_URL = "https://raw.githubusercontent.com/3Zzamt0o/HospitalData/main/Hospital%20Data%20(1).xlsx"
|
10 |
-
|
11 |
-
def load_hospital_data():
|
12 |
-
try:
|
13 |
-
# Fetch Excel file from GitHub
|
14 |
-
response = requests.get(GITHUB_EXCEL_URL)
|
15 |
-
response.raise_for_status() # Raise an exception for bad status codes
|
16 |
-
|
17 |
-
# Read Excel file from the response content
|
18 |
-
excel_data = io.BytesIO(response.content)
|
19 |
-
df = pd.read_excel(excel_data)
|
20 |
-
|
21 |
-
print("Successfully loaded data from GitHub")
|
22 |
-
return df
|
23 |
-
except Exception as e:
|
24 |
-
print(f"Error loading data from GitHub: {e}")
|
25 |
-
# Fallback to default data if GitHub fetch fails
|
26 |
-
return pd.DataFrame({
|
27 |
-
'العيادات': ['عيادة الأسنان', 'عيادة الباطنة', 'عيادة العظام', 'عيادة الأطفال'],
|
28 |
-
'الدكتور': ['د. أحمد محمد', 'د. محمد علي', 'د. علي حسن', 'د. سارة أحمد'],
|
29 |
-
'الأيام': ['السبت والاثنين', 'الأحد والثلاثاء', 'الاثنين والأربعاء', 'الثلاثاء والخميس'],
|
30 |
-
'سعر الكشف': ['100 جنيه', '150 جنيه', '120 جنيه', '130 جنيه']
|
31 |
-
})
|
32 |
-
|
33 |
-
# Load the hospital data
|
34 |
-
df = load_hospital_data()
|
35 |
-
|
36 |
-
data = {
|
37 |
-
"السؤال": [
|
38 |
-
"ما هي مواعيد معمل التحاليل ؟", "ما هي المده المستغرقه لعمل التحاليل ؟",
|
39 |
-
"ما هي انواع التحاليل الموجوده و المتوفره ؟", "هل يوجد تحليل صوره دم ؟",
|
40 |
-
"ماهي تحاليل صوره الدم ؟", "ما هي تحاليل السكر ؟", "ما هي تحاليل وظائف الكبد ؟",
|
41 |
-
"ما هي تحاليل وظائف الكلي ؟", "ما هي تحاليل البول و البراز ؟",
|
42 |
-
"ما هي تحاليل الدهون ؟", "ما هي تحاليل الألتهابات و الروماتيزم ؟",
|
43 |
-
"ما هي تحاليل الفيروسات ؟", "ما هي تحاليل الغده الدرقيه ؟",
|
44 |
-
"هل يوجد حشو اسنان ؟", "هل يوجد جهاز سونار ؟", "هل يوجد اجهزه تنفس اصطناعي ؟",
|
45 |
-
"هل يوجد قياس نظر ؟", "ما هي الاجهزه المتاحه في عياده الرمد ؟",
|
46 |
-
"هل يوجد قياس ضغط ؟", "هل يوجد سونار ؟", "هل يوجد رسم قلب ؟"
|
47 |
-
],
|
48 |
-
"الجواب": [
|
49 |
-
"طول ايام الاسبوع من 9 صباحا الي 9 مسائا", "من 24 ساعه الي 48 ساعه",
|
50 |
-
"تحاليل صوره الدم، تحاليل سكر، تحاليل وظائف الكبد، تحاليل وظائف الكلي",
|
51 |
-
"نعم يوجد جميع انواع التحاليل بالمستشفي", "CBC، Hb%(Heamoglobin)، Blood Group، Rh",
|
52 |
-
"Rbs سكر عشوائي، سكر صائم FBS، PPbs سكر فاطر", "ALT(GPT)، AST(GOT)، Billirubin",
|
53 |
-
"Ceatinin الكرياتنين، Urea اليوريا، Urine البول، Uric acid اليوريك اسد",
|
54 |
-
"Urine البول، Stool البراز، جرثومه المعده في البراز", "Cholesterol كوليسترول، Triglyceride الدهون الثلاثيه",
|
55 |
-
"CRP، ASOT", "HCV Ab، GBS Ag، HIC Ab", "Tsh، T3، t4",
|
56 |
-
"نعم يوجد حشو أسنان", "نعم يوجد جهاز سونار", "نعم يوجد أجهزة تنفس إصطناعي",
|
57 |
-
"نعم يوجد قياس نظر", "جهاز قاع العين، جهاز قياس النظر، جهاز الكشف من الحساسية والالتهابات",
|
58 |
-
"نعم يوجد قياس ضغط", "نعم يوجد سونار", "نعم يوجد رسم قلب"
|
59 |
-
]
|
60 |
-
}
|
61 |
-
|
62 |
-
qa_data = pd.DataFrame(data)
|
63 |
-
qa_data.to_csv("qa_data.csv", index=False)
|
64 |
-
|
65 |
-
# Using ArabertV2 model with medical knowledge
|
66 |
-
model_id = "aubmindlab/bert-base-arabertv2"
|
67 |
-
|
68 |
-
import pandas as pd
|
69 |
-
import json
|
70 |
-
from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
|
71 |
-
from sentence_transformers import SentenceTransformer, util
|
72 |
-
import torch
|
73 |
-
from difflib import get_close_matches
|
74 |
-
|
75 |
-
# Medical knowledge base
|
76 |
-
medical_kb = {
|
77 |
-
"الجفاف": "في حالة الشعور بالجفاف: 1- شرب الكثير من السوائل 2- تناول الأطعمة الغنية بالماء 3- تجنب التعرض المباشر للشمس 4- استشارة الطبيب إذا استمرت الأعراض",
|
78 |
-
"الصداع": "لعلاج الصداع: 1- الراحة في مكان هادئ 2- شرب الماء بكثرة 3- تناول مسكن خفيف 4- تجنب الضوء القوي والضوضاء",
|
79 |
-
"التعب": "للتغلب على التعب: 1- الحصول على قسط كافٍ من النوم 2- تناول غذاء صحي متوازن 3- ممارسة تمارين خفيفة 4- تجنب الإجهاد",
|
80 |
-
"الحمى": "عند الإصابة بالحمى: 1- الراحة التامة 2- شرب السوائل بكثرة 3- خفض درجة الحرارة بالكمادات 4- استشارة الطبيب إذا تجاوزت 39 درجة",
|
81 |
-
"السعال": "لتخفيف السعال: 1- شرب الماء الدافئ مع العسل 2- استخدام مرطب الهواء 3- تجنب المثيرات 4- الراحة وتناول الأدوية المناسبة",
|
82 |
-
"ألم المعدة": "لعلاج ألم المعدة: 1- تجنب الأطعمة الدهنية والحارة 2- تناول وجبات خفيفة 3- شرب مشروبات دافئة كاليانسون 4- استشارة الطبيب إذا استمر الألم",
|
83 |
-
"الإسهال": "لعلاج الإسهال: 1- شرب محلول معالجة الجفاف 2- تجنب الأطعمة الدسمة 3- تناول وجبات خفيفة وسهلة الهضم 4- مراجعة الطبيب إذا استمر لأكثر من يومين",
|
84 |
-
"الإمساك": "لعلاج الإمساك: 1- تناول الألياف مثل الخضروات والفواكه 2- شرب الماء بكثرة 3- ممارسة النشاط البدني 4- استخدام ملينات طبيعية إذا لزم الأمر",
|
85 |
-
"الزكام": "لعلاج الزكام: 1- الراحة 2- شرب السوائل الساخنة 3- استخدام بخاخات الأنف 4- تجنب البرد والتيارات الهوائية",
|
86 |
-
"ألم الظهر": "لتخفيف ألم الظهر: 1- تجنب الجلوس لفترات طويلة 2- ممارسة تمارين التمدد 3- استخدام كمادات دافئة 4- مراجعة الطبيب في حال استمرار الألم"
|
87 |
-
}
|
88 |
-
|
89 |
-
|
90 |
-
print("Loading Arabic model and tokenizer...")
|
91 |
-
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
92 |
-
model = AutoModelForSequenceClassification.from_pretrained(model_id)
|
93 |
-
|
94 |
-
embedder = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
95 |
-
passages = []
|
96 |
-
for _, row in df.iterrows():
|
97 |
-
if pd.isna(row['الدكتور']) and pd.isna(row['العيادات']):
|
98 |
-
continue
|
99 |
-
text = f"العيادة: {row['العيادات']}، الدكتور: {row['الدكتور']}، الأيام: {row['الأيام']}، السعر: {row['سعر الكشف']}"
|
100 |
-
passages.append(text)
|
101 |
-
|
102 |
-
corpus_embeddings = embedder.encode(passages, convert_to_tensor=True)
|
103 |
-
|
104 |
-
|
105 |
-
def answer_question_from_excel(user_question, k=3):
|
106 |
-
# First check if it's a medical symptom question
|
107 |
-
for symptom, advice in medical_kb.items():
|
108 |
-
if symptom in user_question or any(word in user_question for word in ["اشعر", "اعاني", "الم", "وجع", "مريض"]):
|
109 |
-
return f"� **الرد:** {advice}\n\n⚠️ **ملاحظة هامة:** هذه النصائح عامة، يرجى استشارة الطبيب للتشخيص الدقيق."
|
110 |
-
|
111 |
-
# Then check in qa_data for lab tests and equipment
|
112 |
-
qa_answer = check_lab_or_radiology(user_question)
|
113 |
-
if qa_answer:
|
114 |
-
return f"💡 **الرد:** {qa_answer}"
|
115 |
-
|
116 |
-
# Check for specific clinic questions
|
117 |
-
|
118 |
-
|
119 |
-
# Define clinic keywords with variations
|
120 |
-
clinic_mapping = {
|
121 |
-
"عيادة الباطنة": ["باطنة", "الباطنة", "باطنه"],
|
122 |
-
"عيادة الجلدية": ["جلد", "جلدية", "الجلدية"],
|
123 |
-
"عيادة المسالك": ["مسالك", "المسالك", "مسالك بولية"],
|
124 |
-
"عيادة النساء": ["نسا", "نساء", "توليد", "ولادة"],
|
125 |
-
"عيادة الأنف والأذن": ["انف", "اذن", "حنجرة", "الأنف", "الأذن"],
|
126 |
-
"عيادة الرمد": ["رمد", "الرمد", "عيون", "العيون"],
|
127 |
-
"عيادة الأسنان": ["اسنان", "الاسنان", "الأسنان", "سنان"],
|
128 |
-
"الحضانة": ["حضان", "حضانة", "الحضانة", "حضانات"],
|
129 |
-
"المعمل": ["معمل", "المعمل", "تحاليل", "مختبر"],
|
130 |
-
"عيادة الأطفال": ["اطفال", "الاطفال", "الأطفال"],
|
131 |
-
"عيادة الجهاز الهضمي": ["هضم", "هضمي", "الجهاز الهضمي"],
|
132 |
-
"عيادة التخاطب": ["تخاطب", "التخاطب", "نطق"]
|
133 |
-
}
|
134 |
-
|
135 |
-
# Check for specific clinic mentions
|
136 |
-
target_clinic = None
|
137 |
-
for clinic, keywords in clinic_mapping.items():
|
138 |
-
if any(keyword in user_question for keyword in keywords):
|
139 |
-
target_clinic = clinic
|
140 |
-
break
|
141 |
-
|
142 |
-
# If a specific clinic is mentioned, show all information for that clinic
|
143 |
-
if target_clinic:
|
144 |
-
clinic_info = [p for p in passages if target_clinic in p]
|
145 |
-
if clinic_info:
|
146 |
-
return f" **الرد:** \n" + "\n".join(clinic_info)
|
147 |
-
|
148 |
-
# For questions about doctors or clinics
|
149 |
-
if any(word in user_question for word in ["عيادة", "عيادات", "دكتور", "دكاترة", "دك��ورة", "مواعيد"]):
|
150 |
-
# Check if asking about a specific type of clinic
|
151 |
-
for clinic_type in ["الأسنان", "الباطنة", "الجلدية", "المسالك", "النساء", "الأنف", "الرمد", "الحضانة", "المعمل", "الأطفال", "الجهاز", "التخاطب"]:
|
152 |
-
if clinic_type in user_question:
|
153 |
-
clinic_info = [p for p in passages if clinic_type in p]
|
154 |
-
if clinic_info:
|
155 |
-
return f" **الرد:** \n" + "\n".join(clinic_info)
|
156 |
-
|
157 |
-
# If no specific clinic type mentioned, return all clinic information
|
158 |
-
return f"💡 **الرد:** \n" + "\n".join([p for p in passages if 'عيادة' in p])
|
159 |
-
|
160 |
-
# If no specific clinic found, check all hospital data
|
161 |
-
question_embedding = embedder.encode(user_question, convert_to_tensor=True)
|
162 |
-
hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=k)
|
163 |
-
matched_passages = [passages[hit['corpus_id']] for hit in hits[0]]
|
164 |
-
|
165 |
-
if matched_passages:
|
166 |
-
hospital_info = "\n".join(matched_passages)
|
167 |
-
return f"💡 **الرد:** {hospital_info}"
|
168 |
-
|
169 |
-
# If no match found in any source, return a helpful message
|
170 |
-
return f"💡 **الرد:** عذراً، لا يمكنني العثور على إجابة محددة لسؤالك. يرجى استشارة الطبيب للحصول على المشورة الطبية المناسبة."
|
171 |
-
|
172 |
-
|
173 |
-
def check_lab_or_radiology(question):
|
174 |
-
keywords = ['أشعة', 'تحاليل', 'رنين', 'سونار', 'تحليل', 'مختبر', 'معمل']
|
175 |
-
if any(word in question for word in keywords):
|
176 |
-
matches = get_close_matches(question, qa_data['السؤال'].tolist(), n=1, cutoff=0.5)
|
177 |
-
if matches:
|
178 |
-
matched_q = matches[0]
|
179 |
-
answer = qa_data[qa_data['السؤال'] == matched_q]['الجواب'].values[0]
|
180 |
-
return answer
|
181 |
-
return None
|
182 |
-
def answer_question(user_question):
|
183 |
-
answer = check_lab_or_radiology(user_question)
|
184 |
-
if answer:
|
185 |
-
return answer
|
186 |
-
return answer_question_from_excel(user_question)
|
187 |
-
|
188 |
-
#print(answer_question("مين دكاترة الأطفال وميعادهم؟"))
|
189 |
-
#print(answer_question("أريد معرفة تفاصيل التحليل الطبي"))
|
190 |
-
|
191 |
-
#print(answer_question("اشعر ب جفاف و همدان ف جسمي ماذا افعل"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|