Spaces:
Runtime error
Runtime error
Upload 3 files
Browse files- FINAL_CODE.py +191 -0
- app.py +33 -64
- requirements.txt +10 -1
FINAL_CODE.py
ADDED
@@ -0,0 +1,191 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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("اشعر ب جفاف و همدان ف جسمي ماذا افعل"))
|
app.py
CHANGED
@@ -1,64 +1,33 @@
|
|
1 |
-
import
|
2 |
-
from
|
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 |
-
temperature=temperature,
|
35 |
-
top_p=top_p,
|
36 |
-
):
|
37 |
-
token = message.choices[0].delta.content
|
38 |
-
|
39 |
-
response += token
|
40 |
-
yield response
|
41 |
-
|
42 |
-
|
43 |
-
"""
|
44 |
-
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
45 |
-
"""
|
46 |
-
demo = gr.ChatInterface(
|
47 |
-
respond,
|
48 |
-
additional_inputs=[
|
49 |
-
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
50 |
-
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
51 |
-
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
52 |
-
gr.Slider(
|
53 |
-
minimum=0.1,
|
54 |
-
maximum=1.0,
|
55 |
-
value=0.95,
|
56 |
-
step=0.05,
|
57 |
-
label="Top-p (nucleus sampling)",
|
58 |
-
),
|
59 |
-
],
|
60 |
-
)
|
61 |
-
|
62 |
-
|
63 |
-
if __name__ == "__main__":
|
64 |
-
demo.launch()
|
|
|
1 |
+
from flask import Flask, request, jsonify
|
2 |
+
from flask_cors import CORS
|
3 |
+
from FINAL_CODE import answer_question
|
4 |
+
app = Flask(__name__)
|
5 |
+
CORS(app)
|
6 |
+
|
7 |
+
@app.route('/api/ask', methods=['POST'])
|
8 |
+
def ask_question():
|
9 |
+
try:
|
10 |
+
data = request.get_json()
|
11 |
+
if not data or 'question' not in data:
|
12 |
+
return jsonify({
|
13 |
+
'success': False,
|
14 |
+
'error': 'Question is required'
|
15 |
+
}), 400
|
16 |
+
|
17 |
+
question = data['question']
|
18 |
+
answer = answer_question(question)
|
19 |
+
|
20 |
+
return jsonify({
|
21 |
+
'success': True,
|
22 |
+
'question': question,
|
23 |
+
'answer': answer
|
24 |
+
})
|
25 |
+
except Exception as e:
|
26 |
+
return jsonify({
|
27 |
+
'success': False,
|
28 |
+
'error': str(e)
|
29 |
+
}), 500
|
30 |
+
|
31 |
+
if __name__ == '__main__':
|
32 |
+
# Make sure to run with host='0.0.0.0' to be accessible externally if needed
|
33 |
+
app.run(host='0.0.0.0', port=5000, debug=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
requirements.txt
CHANGED
@@ -1 +1,10 @@
|
|
1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
flask==2.3.3
|
2 |
+
pyodbc==4.0.39
|
3 |
+
flask-cors==4.0.0
|
4 |
+
python-dotenv==1.0.0
|
5 |
+
pandas==2.0.3
|
6 |
+
transformers==4.35.2
|
7 |
+
sentence-transformers==2.2.2
|
8 |
+
torch==2.1.1
|
9 |
+
requests==2.31.0
|
10 |
+
openpyxl==3.1.2
|