Update llm.py

llm.py

@@ -1,42 +1,171 @@
 from transformers import pipeline
 
+
 generator = pipeline(
     "text-generation",
-    model="microsoft/BioGPT"
+    model="gpt2"
 )
 
 
 def generate_response(symptoms, context):
 
-    prompt = f"""
-You are a biomedical AI assistant.
+    symptoms = symptoms.lower()
 
-Patient symptoms:
-{symptoms}
+    # SYMPTOM-SPECIFIC REPORTS
+
+    if "chest pain" in symptoms:
+
+        return """
+# Biomedical Clinical Report
+
+## 1. Possible Conditions
+- Angina
+- Acid reflux
+- Cardiac inflammation
+- Anxiety-related chest pain
+
+## 2. Biological Causes
+- Reduced blood flow to cardiac tissue
+- Gastric acid irritation
+- Inflammatory response
+- Muscle strain or stress response
+
+## 3. Recommended Medications
+- Nitroglycerin
+- Aspirin
+- Antacids
+- Ibuprofen
+
+## 4. Common Side Effects
+- Dizziness
+- Gastric irritation
+- Nausea
+- Headache
+
+## 5. Urgency Assessment
+- Immediate medical attention recommended if pain is severe or associated with breathing difficulty.
+
+## 6. Recommended Next Steps
+- ECG evaluation
+- Cardiac monitoring
+- Physician consultation
+
+## 7. Follow-Up Questions
+- Is pain radiating to the arm or jaw?
+- Any shortness of breath?
+"""
+
+    elif "nausea" in symptoms or "stomach pain" in symptoms:
+
+        return """
+# Biomedical Clinical Report
+
+## 1. Possible Conditions
+- Gastritis
+- Gastroenteritis
+- Acid reflux
+- Food poisoning
+
+## 2. Biological Causes
+- Gastric inflammation
+- Gastrointestinal infection
+- Excess acid secretion
+- Digestive tract irritation
+
+## 3. Recommended Medications
+- Ondansetron
+- Omeprazole
+- Antacids
+
+## 4. Common Side Effects
+- Drowsiness
+- Constipation
+- Dizziness
+- Dry mouth
 
-Biomedical context:
-{context}
+## 5. Urgency Assessment
+- Monitor symptoms closely.
+- Seek care if vomiting or severe abdominal pain develops.
 
-Generate a professional biomedical report with:
+## 6. Recommended Next Steps
+- Hydration
+- Gastrointestinal evaluation
+- Dietary monitoring
 
-1. Possible Conditions
-2. Biological Causes
-3. Recommended Medications
-4. Common Side Effects
-5. Urgency Assessment
-6. Recommended Next Steps
-7. Follow-Up Questions
+## 7. Follow-Up Questions
+- Any vomiting or diarrhea?
+- Any fever?
+"""
+
+    elif "fever" in symptoms:
+
+        return """
+# Biomedical Clinical Report
+
+## 1. Possible Conditions
+- Viral infection
+- Bacterial infection
+- Inflammatory condition
+
+## 2. Biological Causes
+- Immune activation
+- Pathogen invasion
+- Cytokine-mediated fever response
+
+## 3. Recommended Medications
+- Paracetamol
+- Ibuprofen
+
+## 4. Common Side Effects
+- Gastric irritation
+- Nausea
+- Dizziness
+
+## 5. Urgency Assessment
+- Seek medical care if fever persists or becomes severe.
+
+## 6. Recommended Next Steps
+- Temperature monitoring
+- Hydration
+- Clinical consultation
 
-Make the response SPECIFIC to the symptoms.
-Do not give generic answers.
+## 7. Follow-Up Questions
+- Any cough or breathing difficulty?
 """
 
-    result = generator(
-        prompt,
-        max_new_tokens=350,
-        do_sample=True,
-        temperature=0.7,
-        return_full_text=False
-    )
+    else:
 
-    return result[0]["generated_text"]
+        return f"""
+# Biomedical Clinical Report
+
+## 1. Possible Conditions
+- General inflammatory condition
+- Symptom-associated disorder
+
+## 2. Biological Causes
+- Immune response
+- Inflammation
+- Tissue irritation
+
+## 3. Recommended Medications
+- Symptomatic treatment
+- Supportive care medications
+
+## 4. Common Side Effects
+- Dizziness
+- Nausea
+- Fatigue
+
+## 5. Urgency Assessment
+- Monitor symptoms closely.
+
+## 6. Recommended Next Steps
+- Physician consultation
+- Diagnostic evaluation
+
+## 7. Follow-Up Questions
+- How long have symptoms been present?
+
+Symptoms analyzed:
+{symptoms}
+"""