Update app.py

app.py

@@ -825,6 +825,98 @@ def predict_and_store_freight(state, *args):
         state.freight_predictions.append(result)
     return result
 
+
+CUSTOM_CSS = """
+/* Horizontal tabs layout */
+.tabs {
+    display: flex !important;
+    flex-direction: column !important;
+    gap: 1rem !important;
+}
+
+.tabs > .tab-nav {
+    display: flex !important;
+    flex-direction: row !important;
+    gap: 0.5rem !important;
+    padding: 0.5rem !important;
+    background: #f8f9fa !important;
+    border-radius: 8px !important;
+    border-bottom: 2px solid #e9ecef !important;
+}
+
+.tabs > .tab-nav > button {
+    flex: 1 !important;
+    padding: 0.75rem 1rem !important;
+    border-radius: 6px !important;
+    border: none !important;
+    background: transparent !important;
+    color: #495057 !important;
+    font-weight: 500 !important;
+    transition: all 0.2s ease !important;
+}
+
+.tabs > .tab-nav > button.selected {
+    background: white !important;
+    color: #228be6 !important;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
+}
+
+.tab-content {
+    padding: 1.5rem !important;
+    background: white !important;
+    border-radius: 8px !important;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.05) !important;
+}
+
+/* Additional styling */
+.main-header {
+    background: linear-gradient(135deg, #0061a8 0%, #003459 100%);
+    padding: 2rem;
+    color: white;
+    border-radius: 8px;
+    margin-bottom: 2rem;
+}
+
+.app-title {
+    font-size: 2.5rem !important;
+    margin-bottom: 0.5rem !important;
+}
+
+.app-subtitle {
+    opacity: 0.9;
+    margin-bottom: 1rem !important;
+}
+
+.action-button {
+    background: #228be6 !important;
+    border-radius: 6px !important;
+    transition: all 0.2s ease !important;
+}
+
+.action-button:hover {
+    transform: translateY(-2px) !important;
+    box-shadow: 0 4px 8px rgba(34, 139, 230, 0.2) !important;
+}
+
+.file-upload {
+    border: 2px dashed #e9ecef !important;
+    border-radius: 8px !important;
+    padding: 1rem !important;
+}
+
+.result-box {
+    background: #f8f9fa !important;
+    border-radius: 6px !important;
+    padding: 1rem !important;
+}
+
+.chart-container {
+    background: white !important;
+    border-radius: 8px !important;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.05) !important;
+}
+"""
+
 def create_interface():
     """Create Gradio interface with enhanced UI"""
     state = SupplyChainState()
@@ -841,167 +933,102 @@ def create_interface():
         # Main Content Tabs
         with gr.Tabs() as tabs:
             # Data Upload Tab
-           with gr.Tab("📊 Data Upload", elem_classes="tab-content"):
-            with gr.Row():
-                with gr.Column(scale=1):
-                    sales_data_upload = gr.File(
-                        file_types=[".xlsx", ".xls", ".csv"],  # Allow all formats
-                        label="📈 Sales Data (Excel or CSV)",
-                        elem_classes="file-upload"
-                    )
-                    gr.Markdown("*Upload sales data in Excel (.xlsx, .xls) or CSV format*", elem_classes="file-instructions")
+            with gr.Tab("📊 Data Upload", elem_classes="tab-content"):
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        sales_data_upload = gr.File(
+                            file_types=[".xlsx", ".xls", ".csv"],
+                            label="📈 Sales Data (Excel or CSV)",
+                            elem_classes="file-upload"
+                        )
+                        gr.Markdown("*Upload sales data in Excel (.xlsx, .xls) or CSV format*", elem_classes="file-instructions")
                     
-                with gr.Column(scale=1):
-                    supplier_data_upload = gr.File(
-                        file_types=[".xlsx", ".xls", ".csv"],
-                        label="🏭 Supplier Data (Excel or CSV)",
-                        elem_classes="file-upload"
-                    )
-                    gr.Markdown("*Upload supplier data in Excel (.xlsx, .xls) or CSV format*", elem_classes="file-instructions")
-            
-            with gr.Row():
-                text_input_area = gr.Textbox(
-                    label="📝 Additional Context",
-                    placeholder="Add market updates, news, or other relevant information...",
-                    lines=5
-                )
-            
-            with gr.Row():
-                upload_status = gr.Textbox(
-                    label="Status",
-                    elem_classes="status-box"
-                )
-                upload_button = gr.Button(
-                    "🔄 Process Data",
-                    variant="primary",
-                    elem_classes="action-button"
-                )
-
-            # Add file format instructions
-            with gr.Row():
-                gr.Markdown("""
-                💡 **Supported File Formats**:
-                - Excel files (.xlsx, .xls)
-                - CSV files (.csv)
+                    with gr.Column(scale=1):
+                        supplier_data_upload = gr.File(
+                            file_types=[".xlsx", ".xls", ".csv"],
+                            label="🏭 Supplier Data (Excel or CSV)",
+                            elem_classes="file-upload"
+                        )
+                        gr.Markdown("*Upload supplier data in Excel (.xlsx, .xls) or CSV format*", elem_classes="file-instructions")
                 
-                Make sure your files contain the required columns for analysis.
-                """, elem_classes="file-format-help")
-            
-            # Analysis Tab
-        with gr.Tab("🔍 Analysis", elem_classes="tab-content"):
-            with gr.Row():
-                gr.Markdown("### Select Analysis Types")
-            
-            with gr.Row():
-                analysis_options = gr.CheckboxGroup(
-                    choices=[
-                        "📈 Demand Forecasting",
-                        "⚠️ Risk Assessment",
-                        "📦 Inventory Optimization",
-                        "🤝 Supplier Performance",
-                        "🌿 Sustainability Analysis"
-                    ],
-                    label="Choose analyses to perform",
-                    value=[]  # Initialize with empty selection
-                )
-            
-            
-            analyze_button = gr.Button(
-                "🚀 Run Analysis",
-                variant="primary",
-                elem_classes="action-button"
-            )
-            
-            with gr.Row():
-                with gr.Column(scale=2):
-                    analysis_output = gr.Textbox(
-                        label="Analysis Results",
-                        elem_classes="result-box"
-                    )
-                with gr.Column(scale=3):
-                    plot_output = gr.Plot(
-                        label="Visualization",
-                        elem_classes="chart-container"
+                with gr.Row():
+                    text_input_area = gr.Textbox(
+                        label="📝 Additional Context",
+                        placeholder="Add market updates, news, or other relevant information...",
+                        lines=5
                     )
-            
-            raw_output = gr.Textbox(
-                label="Processing Status",
-                elem_classes="status-box"
-            )
-            
-            # Freight Cost Prediction Tab
-        with gr.Tab("💰 Cost Prediction", elem_classes="tab-content"):
-            with gr.Row():
-                shipment_mode = gr.Dropdown(
-                    choices=["✈️ Air", "🚢 Ocean", "🚛 Truck"],
-                    label="Transport Mode",
-                    value="✈️ Air"
-                )
-            
-            with gr.Row():
-                with gr.Column():
-                    weight = gr.Slider(
-                        label="📦 Weight (kg)",
-                        minimum=1,
-                        maximum=10000,
-                        step=1,
-                        value=1000
+                
+                with gr.Row():
+                    upload_status = gr.Textbox(
+                        label="Status",
+                        elem_classes="status-box"
                     )
-                with gr.Column():
-                    line_item_value = gr.Slider(
-                        label="💵 Item Value (USD)",
-                        minimum=1,
-                        maximum=1000000,
-                        step=1,
-                        value=10000
+                    upload_button = gr.Button(
+                        "🔄 Process Data",
+                        variant="primary",
+                        elem_classes="action-button"
                     )
-                with gr.Column():
-                    cost_per_kg = gr.Slider(
-                        label="💰 Cost per kg (USD)",
-                        minimum=0,
-                        maximum=500,
-                        step=0.1,
-                        value=50
+
+            # Analysis Tab
+            with gr.Tab("🔍 Analysis", elem_classes="tab-content"):
+                with gr.Row():
+                    analysis_options = gr.CheckboxGroup(
+                        choices=[
+                            "📈 Demand Forecasting",
+                            "⚠️ Risk Assessment",
+                            "📦 Inventory Optimization",
+                            "🤝 Supplier Performance",
+                            "🌿 Sustainability Analysis"
+                        ],
+                        label="Choose analyses to perform",
+                        value=[]
                     )
                 
-            # Mode-specific inputs
-            with gr.Row(visible=False) as air_inputs:
-                air_charter_weight = gr.Slider(
-                    label="Air Charter Weight",
-                    minimum=0,
-                    maximum=10000
-                )
-                air_charter_value = gr.Slider(
-                    label="Air Charter Value",
-                    minimum=0,
-                    maximum=1000000
-                )
-            
-            with gr.Row(visible=False) as ocean_inputs:
-                ocean_weight = gr.Slider(
-                    label="Ocean Weight",
-                    minimum=0,
-                    maximum=10000
-                )
-                ocean_value = gr.Slider(
-                    label="Ocean Value",
-                    minimum=0,
-                    maximum=1000000
-                )
-            
-            with gr.Row(visible=False) as truck_inputs:
-                truck_weight = gr.Slider(
-                    label="Truck Weight",
-                    minimum=0,
-                    maximum=10000
-                )
-                truck_value = gr.Slider(
-                    label="Truck Value",
-                    minimum=0,
-                    maximum=1000000
+                analyze_button = gr.Button(
+                    "🚀 Run Analysis",
+                    variant="primary",
+                    elem_classes="action-button"
                 )
-            
-            with gr.Row():
+                
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        analysis_output = gr.Textbox(
+                            label="Analysis Results",
+                            elem_classes="result-box"
+                        )
+                    with gr.Column(scale=3):
+                        plot_output = gr.Plot(
+                            label="Visualization",
+                            elem_classes="chart-container"
+                        )
+
+            # Cost Prediction Tab
+            with gr.Tab("💰 Cost Prediction", elem_classes="tab-content"):
+                with gr.Row():
+                    shipment_mode = gr.Dropdown(
+                        choices=["✈️ Air", "🚢 Ocean", "🚛 Truck"],
+                        label="Transport Mode",
+                        value="✈️ Air"
+                    )
+                
+                with gr.Row():
+                    with gr.Column():
+                        weight = gr.Slider(
+                            label="📦 Weight (kg)",
+                            minimum=1,
+                            maximum=10000,
+                            step=1,
+                            value=1000
+                        )
+                    with gr.Column():
+                        line_item_value = gr.Slider(
+                            label="💵 Item Value (USD)",
+                            minimum=1,
+                            maximum=1000000,
+                            step=1,
+                            value=10000
+                        )
+                
                 predict_button = gr.Button(
                     "🔍 Calculate Cost",
                     variant="primary",
@@ -1011,14 +1038,13 @@ def create_interface():
                     label="Predicted Cost (USD)",
                     elem_classes="result-box"
                 )
-            
+
             # Chat Tab
             with gr.Tab("💬 Chat", elem_classes="tab-content"):
                 chatbot = gr.Chatbot(
                     label="Chat History",
                     elem_classes="chat-container",
-                    height=400,
-                    type="messages"
+                    height=400
                 )
                 with gr.Row():
                     msg = gr.Textbox(
@@ -1032,7 +1058,7 @@ def create_interface():
                         scale=1,
                         elem_classes="action-button"
                     )
-            
+
             # Report Tab
             with gr.Tab("📑 Report", elem_classes="tab-content"):
                 report_button = gr.Button(
@@ -1043,11 +1069,7 @@ def create_interface():
                 report_download = gr.File(
                     label="Download Report"
                 )
-        
-        # Footer
-        with gr.Row(elem_classes="footer"):
-            gr.Markdown("Designed and Developed by Aditya Ratan © 2025")
-        
+
         # Event Handlers
         def update_mode_inputs(mode):
             return {
@@ -1062,27 +1084,15 @@ def create_interface():
             outputs=[upload_status]
         )
         
-        # Update the analyze button click handler
         analyze_button.click(
             fn=lambda choices, sales, supplier, text: run_analyses(state, choices, sales, supplier, text),
             inputs=[analysis_options, sales_data_upload, supplier_data_upload, text_input_area],
             outputs=[analysis_output, plot_output, raw_output]
         )
 
-        
-        shipment_mode.change(
-            fn=update_mode_inputs,
-            inputs=[shipment_mode],
-            outputs=[air_inputs, ocean_inputs, truck_inputs]
-        )
-        
         predict_button.click(
-            fn=lambda *args: predict_and_store_freight(state, *args),
-            inputs=[
-                weight, line_item_value, cost_per_kg,
-                shipment_mode, air_charter_weight, ocean_weight, truck_weight,
-                air_charter_value, ocean_value, truck_value
-            ],
+            fn=lambda *args: predict_freight_cost(state, *args),
+            inputs=[weight, line_item_value, shipment_mode],
             outputs=[freight_result]
         )
         
@@ -1090,19 +1100,14 @@ def create_interface():
             fn=lambda message: chat_with_navigator(state, message),
             inputs=[msg],
             outputs=[chatbot]
-        ).then(
-            fn=lambda: "",
-            outputs=[msg]
         )
         
         report_button.click(
-            fn=lambda options: generate_pdf_report(state, options),
-            inputs=[analysis_options],
+            fn=lambda: generate_report(state),
             outputs=[report_download]
         )
-    
-    return demo
 
+    return demo
 
 # Update the launch parameters in __main__:
 if __name__ == "__main__":