Update app.py

app.py

@@ -1,30 +1,19 @@
-import streamlit as st
+import gradio as gr
 import pandas as pd
 import os
 from pathlib import Path
 from datetime import datetime
 import random
 from PIL import Image
-import io
 
-# Configure page
-st.set_page_config(
-    page_title="X-Ray Quality Evaluation",
-    page_icon="🏥",
-    layout="wide"
-)
-
-# Initialize session state
-if "session_active" not in st.session_state:
-    st.session_state.session_active = False
-if "image_pairs" not in st.session_state:
-    st.session_state.image_pairs = []
-if "current_index" not in st.session_state:
-    st.session_state.current_index = 0
-if "evaluations" not in st.session_state:
-    st.session_state.evaluations = []
-if "doctor_name" not in st.session_state:
-    st.session_state.doctor_name = ""
+# Global state
+state = {
+    "session_active": False,
+    "image_pairs": [],
+    "current_index": 0,
+    "evaluations": [],
+    "doctor_name": "",
+}
 
 # Helper functions
 def get_image_files(folder_path):
@@ -78,7 +67,7 @@ def load_image(folder, filename):
     try:
         return Image.open(filepath)
     except Exception as e:
-        st.error(f"Error loading image {filename}: {e}")
+        print(f"Error loading image {filename}: {e}")
         return None
 
 def save_evaluations(evaluations, doctor_name):
@@ -95,153 +84,169 @@ def save_evaluations(evaluations, doctor_name):
     df.to_csv(filepath, index=False)
     return filepath, filename
 
-# Main UI
-st.title("🏥 X-Ray Quality Evaluation System")
-st.markdown("**Pairwise Preference Test for Synthetic X-Ray Images**")
+def start_session(doctor_name):
+    """Start a new evaluation session."""
+    if not doctor_name.strip():
+        return (gr.update(visible=False), gr.update(visible=True), "Please enter your name to start a session.")
+    
+    state["doctor_name"] = doctor_name
+    state["image_pairs"] = create_image_pairs()
+    state["session_active"] = True
+    state["current_index"] = 0
+    state["evaluations"] = []
+    
+    if not state["image_pairs"]:
+        return (gr.update(visible=False), gr.update(visible=True), "No image pairs found. Please ensure both roentgen/ and pedisynth/ folders have matching images.")
+    
+    return update_display()
 
-# Sidebar - Session management
-with st.sidebar:
-    st.header("Session Control")
+def get_current_display():
+    """Get current image pair and metadata."""
+    if not state["session_active"] or state["current_index"] >= len(state["image_pairs"]):
+        return None, None, "", ""
     
-    if not st.session_state.session_active:
-        doctor_name = st.text_input("Enter your name:", key="doc_input")
-        
-        if st.button("Start Session", type="primary"):
-            if doctor_name.strip():
-                st.session_state.doctor_name = doctor_name
-                st.session_state.image_pairs = create_image_pairs()
-                st.session_state.session_active = True
-                st.session_state.current_index = 0
-                st.session_state.evaluations = []
-                st.rerun()
-            else:
-                st.error("Please enter your name to start a session.")
+    current_pair = state["image_pairs"][state["current_index"]]
+    
+    # Determine which image goes where
+    if current_pair['roentgen_left']:
+        left_image = load_image("roentgen", current_pair['roentgen'])
+        right_image = load_image("pedisynth", current_pair['pedisynth'])
+        state["current_left_source"] = "roentgen"
+        state["current_right_source"] = "pedisynth"
     else:
-        st.success(f"✓ Session active for: {st.session_state.doctor_name}")
-        progress = st.session_state.current_index / len(st.session_state.image_pairs)
-        st.progress(progress)
-        st.metric("Progress", f"{st.session_state.current_index}/{len(st.session_state.image_pairs)}")
-        
-        if st.button("End Session & Download Results", type="primary"):
-            if st.session_state.evaluations:
-                filepath, filename = save_evaluations(
-                    st.session_state.evaluations,
-                    st.session_state.doctor_name
-                )
-                st.success(f"✓ Results saved: {filename}")
-                
-                # Provide download link
-                with open(filepath, 'r') as f:
-                    csv_data = f.read()
-                st.download_button(
-                    label="Download CSV",
-                    data=csv_data,
-                    file_name=filename,
-                    mime="text/csv"
-                )
-                st.session_state.session_active = False
-                st.rerun()
-            else:
-                st.warning("No evaluations recorded yet.")
+        left_image = load_image("pedisynth", current_pair['pedisynth'])
+        right_image = load_image("roentgen", current_pair['roentgen'])
+        state["current_left_source"] = "pedisynth"
+        state["current_right_source"] = "roentgen"
+    
+    condition = current_pair['condition']
+    progress = f"Image {state['current_index'] + 1} of {len(state['image_pairs'])}"
+    
+    return left_image, right_image, condition, progress
 
-# Main content
-if not st.session_state.session_active:
-    st.info("👈 Enter your name in the sidebar and click 'Start Session' to begin.")
-else:
-    if st.session_state.current_index >= len(st.session_state.image_pairs):
-        st.success("✅ Evaluation complete! All images have been reviewed.")
-        if st.button("End Session & Download Results"):
-            if st.session_state.evaluations:
-                filepath, filename = save_evaluations(
-                    st.session_state.evaluations,
-                    st.session_state.doctor_name
-                )
-                st.success(f"Results saved: {filename}")
-                with open(filepath, 'r') as f:
-                    csv_data = f.read()
-                st.download_button(
-                    label="Download CSV",
-                    data=csv_data,
-                    file_name=filename,
-                    mime="text/csv"
-                )
-                st.session_state.session_active = False
-                st.rerun()
-    else:
-        current_pair = st.session_state.image_pairs[st.session_state.current_index]
-        
-        st.subheading(f"Image {st.session_state.current_index + 1} of {len(st.session_state.image_pairs)}")
-        st.write(f"**Condition:** {current_pair['condition']}")
-        
-        # Display images side by side (blinded - no labels about source)
-        col1, col2 = st.columns(2)
-        
-        # Determine which image goes where
-        if current_pair['roentgen_left']:
-            left_image = load_image("roentgen", current_pair['roentgen'])
-            right_image = load_image("pedisynth", current_pair['pedisynth'])
-            left_source = "roentgen"
-            right_source = "pedisynth"
-        else:
-            left_image = load_image("pedisynth", current_pair['pedisynth'])
-            right_image = load_image("roentgen", current_pair['roentgen'])
-            left_source = "pedisynth"
-            right_source = "roentgen"
-        
-        with col1:
-            st.write("### Option A")
-            if left_image:
-                st.image(left_image, use_column_width=True)
-            else:
-                st.error("Could not load image A")
-        
-        with col2:
-            st.write("### Option B")
-            if right_image:
-                st.image(right_image, use_column_width=True)
-            else:
-                st.error("Could not load image B")
-        
-        # Preference selection
-        st.markdown("---")
-        st.write("**Which image is of better quality?**")
+def update_display():
+    """Update the display with current image pair."""
+    left_img, right_img, condition, progress = get_current_display()
+    
+    if left_img is None:
+        # Session complete
+        state["session_active"] = False
+        return (
+            gr.update(visible=False),
+            gr.update(visible=True),
+            "✅ Evaluation complete! All images have been reviewed."
+        )
+    
+    return (
+        gr.update(visible=True, value=(left_img, right_img)),
+        gr.update(visible=True, value=f"{progress} | Condition: {condition}"),
+        ""
+    )
+
+
+    
+    state["current_index"] += 1
+    
+    return update_display()
+
+def end_session():
+    """End the session and return CSV file."""
+    if not state["evaluations"]:
+        return None, "No evaluations recorded."
+    
+    filepath, filename = save_evaluations(state["evaluations"], state["doctor_name"])
+    
+    with open(filepath, 'rb') as f:
+        csv_data = f.read()
+    
+    state["session_active"] = False
+    state["current_index"] = 0
+    state["evaluations"] = []
+    
+    return filepath, f"✅ Evaluations saved as: {filename}"
+
+# Gradio interface
+with gr.Blocks(title="X-Ray Quality Evaluation") as demo:
+    gr.Markdown("# 🏥 X-Ray Quality Evaluation System")
+    gr.Markdown("**Pairwise Preference Test for Synthetic X-Ray Images**")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### Start Session")
+            doctor_name_input = gr.Textbox(
+                label="Enter your name:",
+                placeholder="Dr. Smith"
+            )
+            start_btn = gr.Button("Start Session", variant="primary", size="lg")
         
-        col_a, col_b, col_equal = st.columns(3)
+        with gr.Column(scale=1):
+            gr.Markdown("### Results")
+            download_btn = gr.Button("End Session & Download Results", variant="primary", size="lg")
+            result_message = gr.Textbox(
+                label="Status",
+                interactive=False,
+                show_label=True
+            )
+    
+    # Image comparison section
+    with gr.Group(visible=False) as eval_section:
+        gr.Markdown("### Compare Images")
         
-        with col_a:
-            if st.button("👈 Prefer Option A", use_container_width=True, key="btn_a"):
-                preferred_source = left_source
-                st.session_state.evaluations.append({
-                    'timestamp': datetime.now().isoformat(),
-                    'condition': current_pair['condition'],
-                    'base_name': current_pair['base_name'],
-                    'preference': preferred_source.capitalize(),
-                    'doctor': st.session_state.doctor_name
-                })
-                st.session_state.current_index += 1
-                st.rerun()
+        progress_text = gr.Textbox(
+            label="Progress",
+            interactive=False,
+            show_label=True
+        )
         
-        with col_b:
-            if st.button("👉 Prefer Option B", use_container_width=True, key="btn_b"):
-                preferred_source = right_source
-                st.session_state.evaluations.append({
-                    'timestamp': datetime.now().isoformat(),
-                    'condition': current_pair['condition'],
-                    'base_name': current_pair['base_name'],
-                    'preference': preferred_source.capitalize(),
-                    'doctor': st.session_state.doctor_name
-                })
-                st.session_state.current_index += 1
-                st.rerun()
+        image_display = gr.Image(
+            label="Options A (left) and B (right)",
+            show_label=True,
+            type="pil"
+        )
         
-        with col_equal:
-            if st.button("↔️ Equal Quality", use_container_width=True, key="btn_equal"):
-                st.session_state.evaluations.append({
-                    'timestamp': datetime.now().isoformat(),
-                    'condition': current_pair['condition'],
-                    'base_name': current_pair['base_name'],
-                    'preference': 'Equal',
-                    'doctor': st.session_state.doctor_name
-                })
-                st.session_state.current_index += 1
-                st.rerun()
+        with gr.Row():
+            pref_a = gr.Button("👈 Prefer Option A", size="lg")
+            pref_b = gr.Button("👉 Prefer Option B", size="lg")
+    
+    error_section = gr.Textbox(
+        label="Message",
+        interactive=False,
+        show_label=True,
+        visible=True
+    )
+    
+    # File download
+    download_file = gr.File(
+        label="CSV Results",
+        visible=False
+    )
+    
+    # Event handlers
+    start_btn.click(
+        start_session,
+        inputs=doctor_name_input,
+        outputs=[eval_section, error_section, result_message]
+    ).then(
+        update_display,
+        outputs=[image_display, progress_text, error_section]
+    )
+    
+    pref_a.click(
+        record_preference,
+        inputs=gr.State("A"),
+        outputs=[image_display, progress_text, error_section]
+    )
+    
+    pref_b.click(
+        record_preference,
+        inputs=gr.State("B"),
+        outputs=[image_display, progress_text, error_section]
+    )
+    
+    download_btn.click(
+        end_session,
+        outputs=[download_file, result_message]
+    )
+
+if __name__ == "__main__":
+    demo.launch()