Create app.py

app.py

@@ -0,0 +1,51 @@
+import streamlit as st
+from transformers import pipeline
+import os
+
+# Initialize the Hugging Face pipeline with the desired model
+model = "Bio-Medical-MultiModal-Llama-3-8B-V1"  # Replace with actual model name from Hugging Face
+diagnosis_pipeline = pipeline("text-generation", model=model)
+
+# Function to get medical diagnosis using the model
+def get_medical_response(patient_name, age, sex, symptoms, xray_mri=None, medical_reports=None):
+    # Prepare the input message with the provided patient details
+    message_content = f"Patient Details:\nName: {patient_name}\nAge: {age}\nSex: {sex}\nSymptoms: {symptoms}"
+    
+    # If X-ray/MRI file is provided, include it
+    if xray_mri:
+        message_content += f"\nX-ray/MRI: {xray_mri}"  # File path or additional info
+    
+    # If medical reports file is provided, include it
+    if medical_reports:
+        message_content += f"\nMedical Reports: {medical_reports}"  # File path or additional info
+    
+    # Use the Hugging Face model to generate a diagnosis response
+    try:
+        result = diagnosis_pipeline(message_content, max_length=300)
+        return result[0]['generated_text']
+    except Exception as e:
+        return f"Error: {str(e)}"  # Return the error message if something goes wrong
+
+# Streamlit UI
+def main():
+    st.title("Medical Diagnosis Assistant")
+    
+    # Collect patient details
+    patient_name = st.text_input("Patient Name")
+    age = st.number_input("Age", min_value=0)
+    sex = st.radio("Sex", options=["Male", "Female", "Other"])
+    symptoms = st.text_area("Medical Symptoms")
+    
+    # Optional file inputs
+    xray_mri = st.file_uploader("Upload X-ray/MRI Image (Optional)", type=["jpg", "jpeg", "png", "dcm", "pdf"])
+    medical_reports = st.file_uploader("Upload Medical Reports (Optional)", type=["pdf", "txt", "docx"])
+
+    if st.button("Submit"):
+        # Get medical diagnosis using the model
+        diagnosis = get_medical_response(patient_name, age, sex, symptoms, xray_mri.name if xray_mri else None, medical_reports.name if medical_reports else None)
+        
+        # Display the response
+        st.text_area("Medical Report Diagnosis", diagnosis, height=300)
+
+if __name__ == "__main__":
+    main()