Update app.py

app.py

@@ -9,11 +9,15 @@ from datetime import datetime
 from datasets import load_dataset
 
 try:
-    pharmkg_dataset = load_dataset("vinven7/PharmKG")
-    print("PharmKG 데이터셋 로드 완료")
+    medical_datasets = {
+        'all_processed': load_dataset("lavita/medical-qa-datasets", "all-processed"),
+        'icliniq': load_dataset("lavita/medical-qa-datasets", "chatdoctor-icliniq"),
+        'healthcaremagic': load_dataset("lavita/medical-qa-datasets", "chatdoctor_healthcaremagic")
+    }
+    print("의료 데이터셋 로드 완료")
 except Exception as e:
-    print(f"PharmKG 데이터셋 로드 실패: {e}")
-    pharmkg_dataset = None
+    print(f"의료 데이터셋 로드 실패: {e}")
+    medical_datasets = None
 
 # 환경 변수 설정
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -162,45 +166,54 @@ def read_uploaded_file(file):
     except Exception as e:
         return f"❌ 파일 읽기 오류: {str(e)}", "error"
 
-
-
-def get_pharmkg_context(query):
-    """PharmKG 데이터셋에서 관련 정보 검색"""
-    if pharmkg_dataset is None:
+def get_medical_context(query):
+    """의료 데이터셋에서 관련 정보 검색"""
+    if medical_datasets is None:
         return ""
     
     try:
-        # 데이터셋에서 관련 정보 검색 로직 구현
         relevant_info = []
         
-        # 약물 정보 검색
-        for item in pharmkg_dataset['train']:
-            if query.lower() in str(item).lower():
-                relevant_info.append(str(item))
+        # 각 데이터셋에서 관련 정보 검색
+        for dataset_name, dataset in medical_datasets.items():
+            for item in dataset['train']:
+                # 질문과 답변에서 관련 정보 검색
+                if 'question' in item and query.lower() in item['question'].lower():
+                    relevant_info.append(f"Q: {item['question']}\nA: {item['answer']}")
+                elif 'answer' in item and query.lower() in item['answer'].lower():
+                    relevant_info.append(f"Q: {item['question']}\nA: {item['answer']}")
+                
+                if len(relevant_info) >= 3:  # 최대 3개까지만 수집
+                    break
         
         if relevant_info:
-            return "\n\nPharmKG 참고 정보:\n" + "\n".join(relevant_info[:3])
+            return "\n\n의료 참고 정보:\n" + "\n---\n".join(relevant_info[:3])
         return ""
     except Exception as e:
-        print(f"PharmKG 검색 오류: {e}")
+        print(f"의료 데이터 검색 오류: {e}")
         return ""
 
 
-# SYSTEM_PREFIX를 전역 변수로 이동
-SYSTEM_PREFIX = """저는 약리학 전문 AI 어시스턴트 'GiniGEN Pharm'입니다. PharmKG 데이터베이스를 기반으로 
-다음과 같은 전문성을 가지고 소통하겠습니다:
-1. 💊 약물 관련 전문 지식 제공
-2. 🔬 약물 상호작용 및 부작용 정보
-3. 🧬 질병-약물 관계 설명
-4. 📊 임상 데이터 기반 분석
-5. ⚕️ 의약품 안전성 정보
+SYSTEM_PREFIX = """저는 의학 전문 AI 어시스턴트 'GiniGEN Medical'입니다. 
+전문 의료 데이터베이스를 기반으로 다음과 같은 전문성을 가지고 소통하겠습니다:
+
+1. 🏥 일반적인 의학 정보 제공
+2. 🔬 증상 및 질병 관련 설명
+3. 🧬 건강 관리 조언
+4. 📊 의학 연구 데이터 해석
+5. ⚕️ 예방 의학 정보
+
 다음 원칙으로 소통하겠습니다:
-1. 🤝 전문적이고 신뢰할 수 있는 정보 제공
-2. 💡 이해하기 쉬운 설명과 함께 과학적 근거 제시
-3. 🎯 질문의 의도를 정확히 파악하여 맞춤형 답변
-4. ⚠️ 의료적 조언이 아닌 정보 제공 목적임을 명시
-5. ✨ 추가 참고자료 및 연구 데이터 제안
-의학적 결정은 반드시 전문 의료진과 상담하시기 바랍니다."""
+1. 🤝 신뢰할 수 있는 의학 정보 제공
+2. 💡 이해하기 쉬운 의학 설명
+3. 🎯 개인별 맞춤 건강 정보
+4. ⚠️ 의료 면책조항 준수
+5. ✨ 과학적 근거 기반 조언
+
+중요 고지사항:
+- 이는 일반적인 정보 제공 목적이며, 전문 의료 상담을 대체할 수 없습니다.
+- 긴급한 의료 상황이나 심각한 증상의 경우 즉시 의료진을 찾아주세요.
+- 모든 치료 결정은 반드시 담당 의료진과 상담 후 결정하시기 바랍니다."""
 
 def chat(message, history, uploaded_file, system_message="", max_tokens=4000, temperature=0.7, top_p=0.9):
     if not message:
@@ -208,7 +221,8 @@ def chat(message, history, uploaded_file, system_message="", max_tokens=4000, te
 
     try:
         # PharmKG 컨텍스트 추가
-        pharmkg_context = get_pharmkg_context(message)
+        pharmkg_context = get_medical_context(message)  # 함수명만 변경        
+
         system_message = SYSTEM_PREFIX + system_message + pharmkg_context
 
         # 파일 업로드 처리
@@ -308,18 +322,17 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", title="GiniGEN 🤖") as demo
                 temperature = gr.Slider(minimum=0, maximum=1, value=0.7, label="창의성 수준 🌡️")
                 top_p = gr.Slider(minimum=0, maximum=1, value=0.9, label="응답 다양성 📈")
 
-    # 예시 질문
+
     gr.Examples(
         examples=[
-            ["안녕하세요! 어떤 도움이 필요하신가요? 🤝"],
-            ["제가 이해하기 쉽게 설명해 주시겠어요? 📚"],
-            ["이 내용을 실제로 어떻게 활용할 수 있을까요? 🎯"],
-            ["추가로 조언해 주실 내용이 있으신가요? ✨"],
-            ["궁금한 점이 더 있는데 여쭤봐도 될까요? 🤔"],
+            ["일반적인 건강 관리 조언을 해주세요. 🏥"],
+            ["고혈압 증상에 대해 설명해주세요. 🔬"],
+            ["건강한 생활습관에 대해 알려주세요. 💪"],
+            ["코로나19 예방수칙을 알려주세요. 🦠"],
+            ["스트레스 관리 방법을 추천해주세요. 🧘‍♀️"],
         ],
         inputs=msg,
-    )
-
+)
     # 대화내용 지우기 버튼에 히스토리 초기화 기능 추가
     def clear_chat():
         chat_history.clear_history()