Update src/streamlit_app.py

src/streamlit_app.py

@@ -18,6 +18,13 @@ import tempfile
 os.environ['STREAMLIT_BROWSER_GATHER_USAGE_STATS'] = 'false'
 os.environ['MPLCONFIGDIR'] = tempfile.gettempdir()
 
+# --- Session state initialization ---
+if 'uploaded_image' not in st.session_state:
+    st.session_state['uploaded_image'] = None
+if 'analysis_complete' not in st.session_state:
+    st.session_state['analysis_complete'] = False
+
+
 # Page configuration
 st.set_page_config(
     page_title="Kolam Design Analyzer",
@@ -362,11 +369,25 @@ except ImportError as e:
 with col1:
     st.markdown("### 📤 Upload Kolam Image")
     
+    #uploaded_file = st.file_uploader(
+        #"Choose a Kolam image...",
+        #type=["png", "jpg", "jpeg"],
+        #help="Upload a clear image of a Kolam design for analysis"
+    #)
     uploaded_file = st.file_uploader(
         "Choose a Kolam image...",
         type=["png", "jpg", "jpeg"],
         help="Upload a clear image of a Kolam design for analysis"
     )
+
+    # Store in session state only if a new file is uploaded
+    if uploaded_file is not None:
+        st.session_state['uploaded_image'] = Image.open(uploaded_file)
+
+    # Display the uploaded image if available
+    if st.session_state['uploaded_image'] is not None:
+        st.image(st.session_state['uploaded_image'], caption="Uploaded Kolam", use_column_width=True)
+
     
     if uploaded_file:
         # Display uploaded image
@@ -374,24 +395,56 @@ with col1:
         st.image(image, caption="Uploaded Kolam", use_column_width=True)
         
         # Analysis button
+        #if st.button("🔍 Analyze Kolam Design", type="primary"):
+            #with st.spinner("Analyzing Kolam design..."):
+                # Process image
+                #original, thresh, edges = analyzer.preprocess_image(
+                    #image, image_size, threshold_value, canny_low, canny_high
+                #)
+                
+                # Detect nodes and edges
+                #nodes = analyzer.detect_nodes(edges, max_corners)
+                #graph_edges = analyzer.detect_edges(edges, nodes, min_line_length)
+                
+                # Build graph
+                #G = analyzer.build_graph(nodes, graph_edges)
+                
+                # Extract features
+                #features = analyzer.extract_graph_features(G)
+                
+                # Store results in session state
+                #st.session_state.analysis_complete = True
+                #st.session_state.original_img = original
+                #st.session_state.thresh_img = thresh
+                #st.session_state.edges_img = edges
+                #st.session_state.nodes = nodes
+                #st.session_state.graph = G
+                #st.session_state.features = features
+                
+                # Generate encryption key
+                #encryption_key = analyzer.generate_encryption_key()
+                #st.session_state.encryption_key = encryption_key
+                
+                #st.success("✅ Analysis completed successfully!")
+    if st.session_state['uploaded_image'] is not None:
         if st.button("🔍 Analyze Kolam Design", type="primary"):
             with st.spinner("Analyzing Kolam design..."):
                 # Process image
                 original, thresh, edges = analyzer.preprocess_image(
-                    image, image_size, threshold_value, canny_low, canny_high
+                    st.session_state['uploaded_image'], image_size, threshold_value, canny_low, canny_high
                 )
-                
+
                 # Detect nodes and edges
                 nodes = analyzer.detect_nodes(edges, max_corners)
                 graph_edges = analyzer.detect_edges(edges, nodes, min_line_length)
-                
+
                 # Build graph
                 G = analyzer.build_graph(nodes, graph_edges)
-                
+
                 # Extract features
                 features = analyzer.extract_graph_features(G)
-                
-                # Store results in session state
+
+                # Save results in session state
                 st.session_state.analysis_complete = True
                 st.session_state.original_img = original
                 st.session_state.thresh_img = thresh
@@ -399,13 +452,14 @@ with col1:
                 st.session_state.nodes = nodes
                 st.session_state.graph = G
                 st.session_state.features = features
-                
+
                 # Generate encryption key
                 encryption_key = analyzer.generate_encryption_key()
                 st.session_state.encryption_key = encryption_key
-                
+
                 st.success("✅ Analysis completed successfully!")
 
+
 with col2:
     st.markdown("### 📊 Analysis Results")