Update app.py

app.py

@@ -1,9 +1,6 @@
-import gradio as gr
+import streamlit as st
 from transformers import pipeline
-#import openai
-import os
-
-#openai.api_key = os.getenv("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")
+from PIL import Image
 
 # Load the pre-trained image classification pipeline
 pipe = pipeline("image-classification", model="ALM-AHME/convnextv2-large-1k-224-finetuned-BreastCancer-Classification-BreakHis-AH-60-20-20")
@@ -31,50 +28,40 @@ def classify_mammogram(img):
 
     return f"Predicted Class: {predicted_label}\nClassification: {classification}\nConfidence: {confidence:.2f}"
 
-            
-    # user_message = {
-    #     "role": "user",
-    #     "content": f""" You are a breast cancer specialist and you have been given a recent breast cancer result of a patient and here are the results:
-    #      Predicted Class: {predicted_label}\nClassification: {classification}\nConfidence: {confidence:.2f}
-
-    #     You have to give a clear explanations about their screening results, potential next steps, and relevant information about breast health based on the scanned mammogram and provide reference to things like "Retrieve Similar Cases," "Explain Diagnostic," "Treatment Recommendations," "Prognosis Analysis," and "Clinical Trial Matching."
+# Streamlit app
+st.title("Enhanced Protocol for Breast Cancer Detection: Streamlining Decisions from Mammograms")
+st.write("Upload an image of a mammogram, and the model will predict whether it is benign or malignant.")
 
-    #     Write a extensive report for this.
-    #     Good Luck!
-        
-    #     """
-    # }
-    # completion = openai.ChatCompletion.create(
-    #     model="gpt-3.5-turbo",
-    #     messages=[user_message]
-    # )
-    # result= completion.choices[0].message.content.strip()
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
-    # return result
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    st.image(image, caption='Uploaded Image.', use_column_width=True)
+    st.write("Classifying...")
+    
+    classification_result = classify_mammogram(image)
+    st.write(classification_result)
 
-# Launch the GUI with the modified output
-iface = gr.Interface(
-    fn=classify_mammogram,
-    inputs=gr.inputs.Image(type="pil", label="Input Mammogram"),
-    outputs=gr.outputs.Textbox(label="Output"),
-    examples=[["1-243.jpg"],["1-285.jpg"],["1-198.jpg"]],
-    title="Enhanced Protocol for Breast Cancer Detection: Streamlining Decisions from Mammograms.",
-    description=(
-        "<div style='display: flex; justify-content: space-between;'>"
-        "<div style='width: 48%;'>"
-        "<p style='font-size: 18px; font-weight: bold;'>Malignant</p>"
-        "<p style='font-size: 16px;'>Non-suspicious: The AI is confident that no suspicious signs are present.<br>"
-        "High risk: The AI is confident that the results are highly suspicious.<br>"
-        "Indeterminate:  The AI is uncertain and not confident in making a definitive classification.</p>"
-        "</div>"
-        "<div style='width: 48%;'>"
-        "<p style='font-size: 18px; font-weight: bold;'>Benign</p>"
-        "<p style='font-size: 16px;'>Suspicious: AI is not that confident. Further Supervision is needed.<br>"
-        "Non-Suspicious: AI is confident that there is nothing to worry about.<br>"
-        "No Risk: AI is confident. No direct supervision is needed.</p>"
-        "</div>"
-        "</div>"
-    )
+st.write(
+    """
+    <div style='display: flex; justify-content: space-between;'>
+        <div style='width: 48%;'>
+            <p style='font-size: 18px; font-weight: bold;'>Malignant</p>
+            <p style='font-size: 16px;'>
+                Non-suspicious: The AI is confident that no suspicious signs are present.<br>
+                High risk: The AI is confident that the results are highly suspicious.<br>
+                Indeterminate: The AI is uncertain and not confident in making a definitive classification.
+            </p>
+        </div>
+        <div style='width: 48%;'>
+            <p style='font-size: 18px; font-weight: bold;'>Benign</p>
+            <p style='font-size: 16px;'>
+                Suspicious: AI is not that confident. Further Supervision is needed.<br>
+                Non-Suspicious: AI is confident that there is nothing to worry about.<br>
+                No Risk: AI is confident. No direct supervision is needed.
+            </p>
+        </div>
+    </div>
+    """, 
+    unsafe_allow_html=True
 )
-
-iface.launch(inbrowser=True)