Update app.py

app.py

@@ -108,79 +108,126 @@ def conversation(qa_chain, message, history):
     new_history = history + [(message, response_answer)]
     return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
 
+
+# [Previous imports remain the same...]
+
 def demo():
-    """Main demo application"""
+    """Main demo application with enhanced layout"""
     theme = gr.themes.Default(
         primary_hue="indigo",
         secondary_hue="blue",
         neutral_hue="slate",
     )
     
-    with gr.Blocks(theme=theme) as demo:
+    # Custom CSS for advanced layout
+    custom_css = """
+        #app-header {
+            text-align: center;
+            padding: 2rem;
+            background: linear-gradient(to right, #1a365d, #2c5282);
+            color: white;
+            margin-bottom: 2rem;
+            border-radius: 0 0 1rem 1rem;
+        }
+        #app-header h1 {
+            font-size: 2.5rem;
+            margin-bottom: 0.5rem;
+            color: white;
+        }
+        #app-header p {
+            font-size: 1.2rem;
+            opacity: 0.9;
+        }
+        .container {
+            max-width: 1400px;
+            margin: 0 auto;
+            padding: 0 1rem;
+        }
+        .features-grid {
+            display: grid;
+            grid-template-columns: repeat(2, 1fr);
+            gap: 1rem;
+            margin-bottom: 2rem;
+        }
+        .feature-card {
+            background: #f8fafc;
+            padding: 1.5rem;
+            border-radius: 0.5rem;
+            border: 1px solid #e2e8f0;
+        }
+        .section-title {
+            font-size: 1.5rem;
+            color: #1a365d;
+            margin-bottom: 1rem;
+            padding-bottom: 0.5rem;
+            border-bottom: 2px solid #e2e8f0;
+        }
+        .control-panel {
+            background: #f8fafc;
+            padding: 1.5rem;
+            border-radius: 0.5rem;
+            margin-bottom: 1rem;
+        }
+        .chat-container {
+            background: white;
+            border-radius: 0.5rem;
+            box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
+        }
+        .reference-panel {
+            background: #f8fafc;
+            padding: 1rem;
+            border-radius: 0.5rem;
+            margin-top: 1rem;
+        }
+    """
+    
+    with gr.Blocks(theme=theme, css=custom_css) as demo:
         vector_db = gr.State()
         qa_chain = gr.State()
         
-        # Header
-        gr.HTML(
-            """
-            <div style='text-align: center; padding: 20px;'>
-                <h1 style='color: #1a365d; margin-bottom: 10px;'>MetroAssist AI - Expert in Metrology Report Analysis</h1>
-                <p style='color: #4a5568; font-size: 1.2em;'>Your intelligent assistant for advanced analysis of metrological documents</p>
-            </div>
-            """
-        )
+        # Enhanced Header
+        with gr.Row(elem_id="app-header"):
+            with gr.Column():
+                gr.HTML(
+                    """
+                    <h1>MetroAssist AI</h1>
+                    <p>Expert System for Metrology Report Analysis</p>
+                    """
+                )
         
-        # Marketing description
-        gr.Markdown(
-            """
-            ### 🔍 Specialized Metrology Analysis
-            
-            MetroAssist AI is a specialized assistant designed to revolutionize metrology report analysis. 
-            Powered by cutting-edge AI technology, it offers:
-
-            * **Precise Analysis**: Detailed interpretation of measurements, calibrations, and compliance
-            * **Intelligent Contextualization**: Deep understanding of metrological standards and norms
-            * **Advanced Technical Support**: Assistance in complex instrument and measurement analyses
-            * **Rapid Processing**: Efficient analysis of multiple technical documents
-            
-            ⚠️ **Security Note**: Your documents are processed with complete security. We do not permanently store confidential data.
-            """
-        )
-
-        with gr.Row():
-            with gr.Column(scale=86):
-                gr.Markdown("### 📥 Step 1: Document Loading and Preparation")
-                gr.Markdown("Upload your metrology reports for expert analysis.")
-                
-                with gr.Row():
-                    document = gr.Files(
-                        label="Upload Metrology Reports (PDF)",
-                        file_count="multiple",
-                        file_types=["pdf"],
-                        interactive=True,
-                    )
-                with gr.Row():
-                    db_btn = gr.Button("Process Documents")
-                with gr.Row():
-                    db_progress = gr.Textbox(
-                        value="Waiting for documents...",
-                        label="Status"
-                    )
-                
-                gr.Markdown("### 🤖 Analysis Engine Configuration")
-                gr.Markdown("Select and configure the AI model to best meet your needs.")
-                
-                with gr.Row():
-                    llm_btn = gr.Radio(
-                        choices=list_llm_simple,
-                        label="Available AI Models",
-                        value=list_llm_simple[0],
-                        type="index"
-                    )
-                
-                with gr.Row():
-                    with gr.Accordion("Advanced Analysis Parameters", open=False):
-                        with gr.Row():
+        # Main Content Container
+        with gr.Row(equal_height=True):
+            # Left Column - Control Panel
+            with gr.Column(scale=1):
+                with gr.Group(visible=True) as control_panel:
+                    gr.Markdown("## Document Processing", elem_classes="section-title")
+                    
+                    # File Upload Section
+                    with gr.Box(elem_classes="control-panel"):
+                        gr.Markdown("### 📄 Upload Documents")
+                        document = gr.Files(
+                            label="Metrology Reports (PDF)",
+                            file_count="multiple",
+                            file_types=["pdf"],
+                        )
+                        db_btn = gr.Button("Process Documents", elem_classes="primary-btn")
+                        db_progress = gr.Textbox(
+                            value="Ready for documents",
+                            label="Processing Status",
+                        )
+                    
+                    # Model Selection Section
+                    with gr.Box(elem_classes="control-panel"):
+                        gr.Markdown("### 🤖 Model Configuration")
+                        llm_btn = gr.Radio(
+                            choices=list_llm_simple,
+                            label="Select AI Model",
+                            value=list_llm_simple[0],
+                            type="index"
+                        )
+                        
+                        # Advanced Parameters
+                        with gr.Accordion("Advanced Settings", open=False):
                             slider_temperature = gr.Slider(
                                 minimum=0.01,
                                 maximum=1.0,
@@ -188,7 +235,6 @@ def demo():
                                 step=0.1,
                                 label="Analysis Precision"
                             )
-                        with gr.Row():
                             slider_maxtokens = gr.Slider(
                                 minimum=128,
                                 maximum=9192,
@@ -196,7 +242,6 @@ def demo():
                                 step=128,
                                 label="Response Length"
                             )
-                        with gr.Row():
                             slider_topk = gr.Slider(
                                 minimum=1,
                                 maximum=10,
@@ -204,90 +249,82 @@ def demo():
                                 step=1,
                                 label="Analysis Diversity"
                             )
-                
-                with gr.Row():
-                    qachain_btn = gr.Button("Initialize Analysis Assistant")
-                with gr.Row():
-                    llm_progress = gr.Textbox(
-                        value="Waiting for initialization...",
-                        label="Assistant Status"
-                    )
-
-            with gr.Column(scale=200):
-                gr.Markdown(
-                    """
-                    ### 💬 Step 2: Expert Consultation and Analysis
-                    Ask questions about your metrology reports. The assistant will provide detailed technical analyses.
-                    
-                    **Suggested questions:**
-                    - Analyze the calibration results of this instrument
-                    - Verify compliance with technical standards
-                    - Identify critical points in measurements
-                    - Compare results with specified limits
-                    - Evaluate measurement uncertainty
-                    - Assess calibration intervals
-                    """
-                )
-                chatbot = gr.Chatbot(
-                    height=505,
-                    label="Metrology Analysis"
-                )
-                
-                with gr.Accordion("Source Document References", open=False):
-                    with gr.Row():
-                        doc_source1 = gr.Textbox(
-                            label="Technical Reference 1",
-                            lines=2
-                        )
-                        source1_page = gr.Number(label="Page")
-                    with gr.Row():
-                        doc_source2 = gr.Textbox(
-                            label="Technical Reference 2",
-                            lines=2
+                        
+                        qachain_btn = gr.Button("Initialize Assistant")
+                        llm_progress = gr.Textbox(
+                            value="Not initialized",
+                            label="Assistant Status"
                         )
-                        source2_page = gr.Number(label="Page")
-                    with gr.Row():
-                        doc_source3 = gr.Textbox(
-                            label="Technical Reference 3",
-                            lines=2
+            
+            # Right Column - Chat Interface
+            with gr.Column(scale=2):
+                with gr.Group() as chat_interface:
+                    gr.Markdown("## Interactive Analysis", elem_classes="section-title")
+                    
+                    # Feature Cards
+                    with gr.Row(equal_height=True) as feature_grid:
+                        with gr.Column():
+                            gr.Markdown(
+                                """
+                                ### 📊 Capabilities
+                                - Calibration Analysis
+                                - Standards Compliance
+                                - Uncertainty Evaluation
+                                """
+                            )
+                        with gr.Column():
+                            gr.Markdown(
+                                """
+                                ### 💡 Best Practices
+                                - Ask specific questions
+                                - Include measurement context
+                                - Specify standards
+                                """
+                            )
+                    
+                    # Chat Interface
+                    with gr.Box(elem_classes="chat-container"):
+                        chatbot = gr.Chatbot(
+                            height=400,
+                            label="Analysis Conversation"
                         )
-                        source3_page = gr.Number(label="Page")
-                
-                with gr.Row():
-                    msg = gr.Textbox(
-                        placeholder="Enter your question about the metrology report...",
-                        label="Your Query"
-                    )
-                with gr.Row():
-                    submit_btn = gr.Button("Submit Query")
-                    clear_btn = gr.ClearButton(
-                        [msg, chatbot],
-                        value="Clear Conversation"
-                    )
-
+                        with gr.Row():
+                            msg = gr.Textbox(
+                                placeholder="Ask about your metrology report...",
+                                label="Query",
+                                scale=4
+                            )
+                            submit_btn = gr.Button("Send")
+                            clear_btn = gr.ClearButton([msg, chatbot], value="Clear")
+                    
+                    # Reference Panel
+                    with gr.Accordion("Document References", open=False, elem_classes="reference-panel"):
+                        with gr.Row():
+                            with gr.Column():
+                                doc_source1 = gr.Textbox(label="Reference 1", lines=2)
+                                source1_page = gr.Number(label="Page")
+                            with gr.Column():
+                                doc_source2 = gr.Textbox(label="Reference 2", lines=2)
+                                source2_page = gr.Number(label="Page")
+                            with gr.Column():
+                                doc_source3 = gr.Textbox(label="Reference 3", lines=2)
+                                source3_page = gr.Number(label="Page")
+        
         # Footer
-        gr.Markdown(
-            """
-            ---
-            ### ℹ️ About MetroAssist AI
-            
-            Developed for metrology professionals, engineers, and technicians who need precise
-            and reliable analysis of technical documents. Our tool uses advanced AI technology
-            to provide valuable insights and support decision-making in metrology.
-            
-            **Specialized Features:**
-            - Detailed analysis of calibration certificates
-            - Interpretation of complex metrological data
-            - Verification of compliance with technical standards
-            - Decision support in metrological processes
-            - Uncertainty analysis and measurement traceability
-            - Quality control and measurement system analysis
-            
-            *Version 1.0 - Updated 2024*
-            """
-        )
+        with gr.Row():
+            gr.Markdown(
+                """
+                ---
+                ### About MetroAssist AI
+                
+                A specialized tool for metrology professionals, providing advanced analysis 
+                of calibration certificates, measurement data, and technical standards compliance.
+                
+                **Version 1.0** | © 2024 MetroAssist AI
+                """
+            )
 
-        # Event handlers
+        # Event Handlers
         db_btn.click(
             initialize_database,
             inputs=[document],