Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,7 +7,6 @@ import os
 # PDF Extraction Function
 # -------------------------------------------------------
 def extract_pdf_table(uploaded_file):
-    """Extract table data from PDF and convert into DataFrame."""
     rows = []
     with pdfplumber.open(uploaded_file) as pdf:
         for page in pdf.pages:
@@ -16,118 +15,161 @@ def extract_pdf_table(uploaded_file):
                 rows.extend(table)
 
     if not rows:
-        return None, "No table detected in PDF."
+        return None, "No table detected in this PDF."
 
-    num_cols = len(rows[0])
-    auto_columns = [f"Column_{i+1}" for i in range(num_cols)]
+    columns = [f"Column_{i+1}" for i in range(len(rows[0]))]
+    df = pd.DataFrame(rows[1:], columns=columns)
+    return df, "Pricelist extracted successfully!"
 
-    df = pd.DataFrame(rows[1:], columns=auto_columns)
-    return df, "PDF extracted successfully!"
 
 # -------------------------------------------------------
-# UI START
+# APP TITLE
 # -------------------------------------------------------
-st.title("📸 Camera Pricelist & Quotation System")
+st.title("📸 CCTV Pricelist & Quotation Generator")
+
+st.write("### 1️⃣ Select what you want to generate:")
 
 # -------------------------------------------------------
-# Load Default PDF from Repo (Optional)
+# MAIN DROPDOWN: SERVICES OR PRODUCTS
 # -------------------------------------------------------
-st.subheader("1️⃣ Pricelist Loading")
-
-PDF_PATH = "Pollo Price List 11.07.25.pdf"  # rename your file to this
-pricelist_df = None
+choice = st.selectbox(
+    "Choose Option",
+    ["Select Option", "Products", "Services"]
+)
 
-if os.path.exists(PDF_PATH):
-    st.success(f"Loaded pricelist from: {PDF_PATH}")
-    with open(PDF_PATH, "rb") as file:
-        pricelist_df, msg = extract_pdf_table(file)
-    st.info(msg)
+st.write("---")
 
-    if pricelist_df is not None:
-        st.dataframe(pricelist_df)
-else:
-    st.warning("No pricelist file found. Upload below instead.")
 
 # -------------------------------------------------------
-# Upload PDF from User
+# SERVICES UI SECTION
 # -------------------------------------------------------
-uploaded_pdf = st.file_uploader("Upload your PDF pricelist", type=["pdf"])
+if choice == "Services":
 
-if uploaded_pdf:
-    pricelist_df, msg = extract_pdf_table(uploaded_pdf)
-    st.success(msg)
-    if pricelist_df is not None:
-        st.dataframe(pricelist_df)
+    st.header("📝 Service Quotation")
 
-st.write("---")
+    service_name = st.text_input("Service Name")
+    service_desc = st.text_area("Service Description")
+    service_cost = st.number_input("Service Price", min_value=0.0)
 
-# -------------------------------------------------------
-# Product Selection UI
-# -------------------------------------------------------
-st.header("2️⃣ Product / Service Selection")
+    area_value = st.text_input("Approx. Area Covered")
+    area_unit = st.radio("Area Unit", ["Marla", "Sq. Yd"], horizontal=True)
 
-service_or_product = st.selectbox("Select Type", ["Services", "Products"])
+    if st.button("Generate Service Quotation"):
+        st.markdown(f"""
+        # 📄 Service Quotation
 
-product_category = st.selectbox(
-    "Select Product Category",
-    ["Camera", "Video Recorder", "POE Switch", "Cables", "Accessories"]
-)
+        ### ▶ Service Details
+        - **Name:** {service_name}
+        - **Description:** {service_desc}
+        - **Cost:** {service_cost}
 
-if product_category == "Camera":
-    camera_type = st.selectbox("Camera Type", ["Analog Camera", "Digital Camera"])
-    placement_type = st.selectbox("Placement Type", ["Outdoor (Bullet)", "Indoor (Dome)"])
-    resolution = st.selectbox("Camera Resolution", ["1MP", "2MP", "4MP", "5MP", "8MP", "4K"])
-    ir_range = st.multiselect("Infrared Range", ["20m", "30m", "50m", "150m", "300m", "500m"])
-    night_vision = st.selectbox("Night Vision / Color Camera", ["Yes", "No"])
-    feature = st.selectbox("Additional Feature", ["None", "Battery", "4G", "Solar", "Heat Sensing", "AI Based"])
-else:
-    camera_type = "N/A"
-    placement_type = "N/A"
-    resolution = "N/A"
-    ir_range = []
-    night_vision = "N/A"
-    feature = "None"
+        ### ▶ Site Details
+        - **Area:** {area_value} {area_unit}
+        """)
 
-st.write("---")
 
 # -------------------------------------------------------
-# Quotation Section
+# PRODUCTS UI SECTION
 # -------------------------------------------------------
-st.header("3️⃣ Quotation Section")
+elif choice == "Products":
+
+    st.subheader("📄 Pricelist Loading")
+
+    # Use your exact file name
+    PDF_PATH = "Pollo Price List 11.07.25.pdf"
+    pricelist_df = None
+
+    # -------------------------------------------------------
+    # AUTO LOAD DEFAULT PDF IF PRESENT
+    # -------------------------------------------------------
+    if os.path.exists(PDF_PATH):
+        st.success(f"Loaded pricelist: {PDF_PATH}")
+        with open(PDF_PATH, "rb") as file:
+            pricelist_df, msg = extract_pdf_table(file)
+        st.info(msg)
+        if pricelist_df is not None:
+            st.dataframe(pricelist_df)
+    else:
+        st.warning("Default pricelist not found. Upload your own below.")
+
+    # -------------------------------------------------------
+    # USER PDF UPLOAD
+    # -------------------------------------------------------
+    uploaded_pdf = st.file_uploader("Upload PDF Pricelist", type=["pdf"])
+
+    if uploaded_pdf:
+        pricelist_df, msg = extract_pdf_table(uploaded_pdf)
+        st.success(msg)
+        if pricelist_df is not None:
+            st.dataframe(pricelist_df)
+
+    st.write("---")
+
+    # -------------------------------------------------------
+    # PRODUCT SELECTION
+    # -------------------------------------------------------
+    st.header("🛒 Product Selection")
+
+    product_category = st.selectbox(
+        "Select Product Category",
+        ["Camera", "Video Recorder", "POE Switch", "Cables", "Accessories"]
+    )
+
+    # ------------------- CAMERA UI --------------------
+    if product_category == "Camera":
+        camera_type = st.selectbox("Camera Type", ["Analog Camera", "Digital Camera"])
+        placement = st.selectbox("Placement", ["Outdoor (Bullet)", "Indoor (Dome)"])
+        resolution = st.selectbox("Resolution", ["1MP", "2MP", "4MP", "5MP", "8MP", "4K"])
+        ir_range = st.multiselect("Infrared Range", ["20m", "30m", "50m", "150m", "300m", "500m"])
+        night_vision = st.selectbox("Night Vision", ["Yes", "No"])
+        feature = st.selectbox("Additional Feature", ["None", "Battery", "4G", "Solar", "AI Based"])
 
-quotation_type = st.selectbox("Quotation System Type", ["Analog System", "Digital System"])
+    else:
+        camera_type = "N/A"
+        placement = "N/A"
+        resolution = "N/A"
+        ir_range = []
+        night_vision = "N/A"
+        feature = "None"
 
-area_value = st.text_input("Approx. Covered Area")
-area_unit = st.radio("Area Unit", ["Marla", "Sq. Yd"], horizontal=True)
+    st.write("---")
 
-# -------------------------------------------------------
-# Quotation Generator
-# -------------------------------------------------------
-def generate_quote():
-    result = f"""
-# 📄 Quotation Summary
-### ▶ Selection Details
-- **Type:** {service_or_product}
-- **Category:** {product_category}
-- **Camera Type:** {camera_type}
-- **Placement:** {placement_type}
-- **Resolution:** {resolution}
-- **IR Range:** {", ".join(ir_range) if ir_range else "None"}
-- **Night Vision:** {night_vision}
-- **Feature:** {feature}
----
-### ▶ System
-- **System Type:** {quotation_type}
-- **Approx. Area:** {area_value} {area_unit}
----
-"""
-    if pricelist_df is not None:
-        result += "### ▶ Pricelist Preview\n"
-        result += pricelist_df.head().to_markdown(index=False)
-    else:
-        result += "_No pricelist loaded._"
+    # -------------------------------------------------------
+    # PRODUCT QUOTATION SECTION
+    # -------------------------------------------------------
+    st.header("📦 Product System Quotation")
+
+    system_type = st.selectbox("System Type", ["Analog System", "Digital System"])
+
+    area_value = st.text_input("Approx. Area Covered")
+    area_unit = st.radio("Area Unit", ["Marla", "Sq. Yd"], horizontal=True)
 
-    return result
+    if st.button("Generate Product Quotation"):
+        ir_display = ", ".join(ir_range) if ir_range else "None"
 
-if st.button("Generate Quotation"):
-    st.markdown(generate_quote())
+        st.markdown(f"""
+        # 📄 Product Quotation
+
+        ### ▶ Product Details
+        - **Category:** {product_category}
+        - **Camera Type:** {camera_type}
+        - **Placement:** {placement}
+        - **Resolution:** {resolution}
+        - **Infrared:** {ir_display}
+        - **Night Vision:** {night_vision}
+        - **Feature:** {feature}
+
+        ### ▶ System Details
+        - **System Type:** {system_type}
+        - **Coverage Area:** {area_value} {area_unit}
+
+        """)
+
+        if pricelist_df is not None:
+            st.write("### ▶ Pricelist Preview")
+            st.dataframe(pricelist_df.head())
+        else:
+            st.info("No pricelist available.")
+
+else:
+    st.info("Please choose **Products** or **Services** to begin.")