Update app.py

app.py

@@ -101,3 +101,56 @@ with tab1:
             file_name=f"Diabetic_Retinopathy_Report_{patient_name.replace(' ', '_')}.pdf",
             mime='application/octet-stream'
         )
+
+with tab2:
+    st.write("### Capture an image using your camera")
+    
+    # Capture image from camera
+    camera_image = st.camera_input("Take a picture")
+    
+    if camera_image is not None:
+        # Open and display the captured image
+        image = Image.open(camera_image)
+        
+        # Preprocess the image
+        img_array = preprocess_image(image)
+        
+        # Prepare inputs for model
+        img_array_pair = [img_array, img_array]  # Model expects two inputs
+        
+        # Make prediction
+        predictions = model.predict(img_array_pair)[0]
+        
+        # Convert predictions to percentages
+        prediction_percentages = predictions * 100
+        
+        # Find the class with the highest probability
+        highest_index = np.argmax(prediction_percentages)
+        predicted_class = class_labels[highest_index]
+        
+        # Display the image and predictions
+        col1, col2 = st.columns([1, 2])  # Set the width ratio between columns
+        
+        # Display image in the first column
+        with col1:
+            st.image(image, caption='Captured Image', width=150)
+        
+        # Display the predictions in the second column
+        with col2:
+            st.write(f"### Predicted Level: **{predicted_class}**")
+            
+            st.write("### Prediction Results")
+            for i, label in enumerate(class_labels):
+                st.progress(int(prediction_percentages[i]))
+                st.write(f"{label}: {prediction_percentages[i]:.2f}%")
+        
+        # Create and download PDF report
+        pdf_output = create_pdf(patient_name, patient_age, predicted_class, prediction_percentages)
+        
+        # Button to download the PDF
+        st.download_button(
+            label="Download Report",
+            data=pdf_output,
+            file_name=f"Diabetic_Retinopathy_Report_{patient_name.replace(' ', '_')}.pdf",
+            mime='application/octet-stream'
+        )