Upload lassa_pred.py

lassa_pred.py

@@ -0,0 +1,106 @@
+import openai
+import pandas as pd
+from sklearn.ensemble import RandomForestClassifier
+import joblib
+import gradio as gr
+
+# Load the trained classifier model
+model = joblib.load('model_pkl')
+
+# OpenAI API key (assumed to be set in Hugging Face environment)
+openai.api_key = 'sk-jtt46VTrnym9B57gkkS_QAJ2E0RYodrEDHodAgliVqT3BlbkFJKooyGzBYbWkWkIt4-MG6ASNt4pMguQs2Mob6kpkaEA'
+
+# Function to simulate the medical assistant's interaction
+def medical_assistant_interaction(pulse_rate, blood_pressure_systolic, blood_pressure_diastolic, temperature_celsius, 
+                                  headache, muscle_pain, sore_throat, nausea_vomiting, diarrhea, joint_pains, cough):
+    try:
+        # Prepare symptom responses
+        symptom_responses = {
+            "Headache": headache,
+            "Muscle Pain": muscle_pain,
+            "Sore Throat": sore_throat,
+            "Nausea and Vomiting": nausea_vomiting,
+            "Diarrhea": diarrhea,
+            "Joint Pains": joint_pains,
+            "Dry Cough or Coughing Blood": cough
+        }
+
+        # Combine responses into a single text input
+        patient_responses = ', '.join([f"{symptom}: {response}" for symptom, response in symptom_responses.items()])
+
+        # Compile sensor data for the classifier
+        features = pd.DataFrame({
+            'Pulse_Rate': [pulse_rate],
+            'Blood_Pressure_Systolic': [blood_pressure_systolic],
+            'Blood_Pressure_Diastolic': [blood_pressure_diastolic],
+            'Temperature_Celsius': [temperature_celsius]
+        })
+
+        # Make a prediction using the trained classifier
+        prediction = model.predict(features)[0]
+
+        # Use OpenAI to make a combined inference
+        response_text = f"""
+        A patient has the following sensor readings:
+        - Pulse Rate: {pulse_rate} bpm
+        - Systolic Blood Pressure: {blood_pressure_systolic} mmHg
+        - Diastolic Blood Pressure: {blood_pressure_diastolic} mmHg
+        - Temperature: {temperature_celsius}°C
+
+        The patient reported the following symptoms: {patient_responses}.
+        
+        Based on these readings and symptoms, what is the likelihood of Lassa fever? Provide additional follow-up questions if necessary.
+        """
+
+        # Call OpenAI's API for a response
+        response = openai.Completion.create(
+            engine="text-davinci-003",
+            prompt=response_text,
+            max_tokens=150
+        )
+        assistant_message = response['choices'][0]['text'].strip()
+
+        # Combine the assistant's message with the model's prediction
+        if prediction == 1:
+            model_inference = "Based on the sensor readings, the classifier suggests a high likelihood of Lassa fever."
+        else:
+            model_inference = "Based on the sensor readings, the classifier suggests a low likelihood of Lassa fever."
+
+        final_inference = f"{assistant_message}\n\n{model_inference}"
+        
+        return final_inference
+
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
+# Gradio interface for interaction
+def gradio_interface(pulse_rate, blood_pressure_systolic, blood_pressure_diastolic, temperature_celsius, 
+                     headache, muscle_pain, sore_throat, nausea_vomiting, diarrhea, joint_pains, cough):
+    return medical_assistant_interaction(
+        pulse_rate, blood_pressure_systolic, blood_pressure_diastolic, temperature_celsius, 
+        headache, muscle_pain, sore_throat, nausea_vomiting, diarrhea, joint_pains, cough
+    )
+
+# Create a Gradio interface
+interface = gr.Interface(
+    fn=gradio_interface,
+    inputs=[
+        gr.Number(label="Pulse Rate (bpm)"),
+        gr.Number(label="Systolic Blood Pressure (mmHg)"),
+        gr.Number(label="Diastolic Blood Pressure (mmHg)"),
+        gr.Number(label="Temperature (Celsius)"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing headaches?"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing muscle pain?"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing a sore throat?"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing nausea and vomiting?"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing diarrhea?"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing joint pains?"),
+        gr.Radio(choices=["Yes", "No"], label="Are you experiencing a dry cough or coughing blood?")
+    ],
+    outputs="text",
+    title="Lassa Fever Medical Assistant",
+    description="This assistant uses sensor data and patient symptom responses to infer the likelihood of Lassa fever."
+)
+
+# Launch the Gradio interface
+interface.launch()