Update app.py

app.py

@@ -10,7 +10,6 @@ from fpdf import FPDF
 def load_data():
     if os.path.exists("data.csv"):
         df = pd.read_csv("data.csv")
-        # Ensure all records have a SerialNo
         if 'SerialNo' not in df.columns:
             df['SerialNo'] = range(1, len(df)+1)
         return df
@@ -18,7 +17,6 @@ def load_data():
 
 # Save data
 def save_data(df):
-    # Ensure SerialNo is continuous
     df['SerialNo'] = range(1, len(df)+1)
     df.to_csv("data.csv", index=False)
 
@@ -37,14 +35,14 @@ def export_as_pdf(customer):
     
     if customer["StitchingHistory"]:
         pdf.cell(0, 10, "Stitching History:", ln=True)
-        for entry in customer["StitchingHistory"].split(" || "):
+        for entry in str(customer["StitchingHistory"]).split(" || "):
             pdf.multi_cell(0, 10, f"- {entry}")
     
     filename = f"Tailor_Record_{customer['SerialNo']}_{customer['Name']}.pdf"
     pdf.output(filename)
     return filename
 
-# Initialize app
+# UI Starts
 st.set_page_config("Tailor Record App", layout="wide")
 st.title("👔 Tailor Record Management System")
 
@@ -56,7 +54,6 @@ with st.form("new_customer"):
     col1, col2 = st.columns(2)
     with col1:
         name = st.text_input("Customer Name*", max_chars=100)
-        # Phone number input that only accepts numbers
         phone = st.text_input("Phone Number*", max_chars=15, key="phone_input")
     with col2:
         sizing_text = st.text_area("Sizing Details*", height=200)
@@ -104,126 +101,3 @@ with st.form("new_customer"):
             save_data(df)
             st.success("Customer record saved successfully.")
             st.rerun()
-
-# View Records
-st.subheader("🔍 Search & Manage Customer Records")
-
-if not df.empty:
-    # Search functionality
-    search_term = st.text_input("Search by Name, Phone, or Serial No:")
-    if search_term:
-        search_mask = (
-            df['Name'].str.contains(search_term, case=False) |
-            df['Phone'].str.contains(search_term) |
-            df['SerialNo'].astype(str).str.contains(search_term)
-        filtered_df = df[search_mask]
-    else:
-        filtered_df = df
-    
-    if filtered_df.empty:
-        st.info("No matching records found.")
-    else:
-        # Display all matching records in a table with select option
-        st.dataframe(
-            filtered_df[['SerialNo', 'Name', 'Phone']],
-            use_container_width=True,
-            hide_index=True
-        )
-        
-        # Select customer to view/edit
-        selected_serial = st.selectbox(
-            "Select customer by Serial No to manage:",
-            filtered_df['SerialNo'],
-            format_func=lambda x: f"{x} - {filtered_df.loc[filtered_df['SerialNo'] == x, 'Name'].values[0]}"
-        )
-        
-        selected_index = df[df['SerialNo'] == selected_serial].index[0]
-        customer = df.loc[selected_index]
-
-        st.markdown(f"### 📄 Details for **{customer['Name']}** (Serial No: {customer['SerialNo']})")
-        st.write(f"**Phone:** {customer['Phone']}")
-        st.write(f"**Sizing Details:** {customer['SizingText']}")
-        
-        if customer["ImagePath"] and os.path.exists(customer["ImagePath"]):
-            try:
-                st.image(customer["ImagePath"], caption="Sizing Image", use_column_width=True)
-            except Exception as e:
-                st.error(f"Error loading image: {e}")
-        
-        if customer["StitchingHistory"]:
-            st.markdown("**🧵 Stitching History:**")
-            for entry in customer["StitchingHistory"].split(" || "):
-                st.markdown(f"- {entry}")
-
-        col1, col2, col3 = st.columns(3)
-        with col1:
-            if st.button("📝 Edit Record"):
-                with st.form("update_form"):
-                    updated_name = st.text_input("Name*", value=customer["Name"])
-                    updated_phone = st.text_input("Phone*", value=customer["Phone"])
-                    updated_sizing = st.text_area("Sizing Details*", value=customer["SizingText"], height=200)
-                    
-                    # Handle existing stitching history
-                    existing_history = customer["StitchingHistory"].split(" || ") if customer["StitchingHistory"] else []
-                    st.markdown("**🧵 Update Stitching History (Optional):**")
-                    new_history_entries = []
-                    with st.expander("Edit History Entries"):
-                        num_entries = st.number_input("Number of entries", min_value=0, max_value=10, step=1, value=len(existing_history))
-                        for i in range(num_entries):
-                            if i < len(existing_history):
-                                date_part, desc_part = existing_history[i].split(" - ", 1) if " - " in existing_history[i] else (existing_history[i], "")
-                                try:
-                                    default_date = datetime.strptime(date_part, "%Y-%m-%d").date()
-                                except:
-                                    default_date = datetime.now().date()
-                                default_desc = desc_part
-                            else:
-                                default_date = datetime.now().date()
-                                default_desc = ""
-                            
-                            date = st.date_input(f"Stitch Date {i+1}", value=default_date, key=f"update_date_{i}")
-                            desc = st.text_input(f"Description (optional) for Stitch {i+1}", value=default_desc, key=f"update_desc_{i}")
-                            new_history_entries.append(f"{date} - {desc}")
-                    
-                    submitted_update = st.form_submit_button("Update Now")
-                    if submitted_update:
-                        if not updated_name or not updated_phone or not updated_phone.isdigit() or not updated_sizing:
-                            st.error("Name, valid phone number, and sizing details are required.")
-                        else:
-                            df.at[selected_index, "Name"] = updated_name.strip()
-                            df.at[selected_index, "Phone"] = updated_phone.strip()
-                            df.at[selected_index, "SizingText"] = updated_sizing.strip()
-                            df.at[selected_index, "StitchingHistory"] = " || ".join(new_history_entries) if new_history_entries else ""
-                            save_data(df)
-                            st.success("Record updated successfully.")
-                            st.rerun()
-
-        with col2:
-            if st.button("🗑️ Delete Record"):
-                st.warning("Are you sure you want to delete this record?")
-                if st.button("✅ Confirm Delete"):
-                    if customer["ImagePath"] and os.path.exists(customer["ImagePath"]):
-                        try:
-                            os.remove(customer["ImagePath"])
-                        except Exception as e:
-                            st.error(f"Error deleting image: {e}")
-                    df = df.drop(index=selected_index).reset_index(drop=True)
-                    save_data(df)
-                    st.success("Record deleted successfully.")
-                    st.rerun()
-
-        with col3:
-            if st.button("📤 Export as PDF"):
-                try:
-                    filename = export_as_pdf(customer.to_dict())
-                    with open(filename, "rb") as f:
-                        st.download_button(
-                            "⬇️ Download PDF",
-                            f,
-                            file_name=os.path.basename(filename),
-                            mime="application/pdf"
-                        )
-                except Exception as e:
-                    st.error(f"Error generating PDF: {e}")
-else:
-    st.info("No customer records found.")