Update app.py

app.py

@@ -164,7 +164,7 @@ def query_router(query, method, retriever):
 
 # Function to reset input and output
 def reset_query_field():
-    return "",""  # Reset only the query input
+    return "", ""  # Reset only the query input
 
 # Function to update the clock
 def update_datetime():
@@ -190,7 +190,6 @@ if os.path.exists(src_logo) and not os.path.exists(dst_logo):
     shutil.copy(src_logo, dst_logo)
     print(f"Copied logo to {dst_logo}")
 
-
 # Main function to create and launch the Gradio interface
 def main():
     # Initialize retriever
@@ -236,81 +235,78 @@ def main():
     }
     """
     
-    logo_path = "Equinix-LOGO.jpeg"  # Make sure this file exists in the Space
-
-    # Then in your UI code:
-    with gr.Row(elem_id="header-container"):
-        with gr.Column():
-            if os.path.exists(dst_logo):
-                gr.HTML(f"""
-                    <div style="text-align:center; background-color:black; padding:15px;">
-                        <img src="/static/Equinix-LOGO.jpeg" alt="Equinix Logo" style="max-height:80px;">
-                    </div>
-                """)
-            else:
-                # Fallback to text logo
-                gr.HTML("""
-                    <div style="text-align:center; background-color:black; padding:15px;">
-                        <div style="color:#FF0000; font-size:24px;">■■■■</div>
-                        <div style="font-size:28px; letter-spacing:5px; margin-top:5px; color:white;">EQUINIX</div>
-                    </div>
-                """)
+    # Create the Gradio interface using Blocks
+    with gr.Blocks(css=custom_css) as demo:
+        with gr.Row(elem_id="header-container"):
+            with gr.Column():
+                if os.path.exists(dst_logo):
+                    gr.HTML(f"""
+                        <div style="text-align:center; background-color:black; padding:15px;">
+                            <img src="/static/Equinix-LOGO.jpeg" alt="Equinix Logo" style="max-height:80px;">
+                        </div>
+                    """)
+                else:
+                    # Fallback to text logo
+                    gr.HTML("""
+                        <div style="text-align:center; background-color:black; padding:15px;">
+                            <div style="color:#FF0000; font-size:24px;">■■■■</div>
+                            <div style="font-size:28px; letter-spacing:5px; margin-top:5px; color:white;">EQUINIX</div>
+                        </div>
+                    """)
+        
+        # Title & Description
+        gr.Markdown("<h1 style='text-align: center; color: black;'>Equinix Chatbot for Automation Team</h1>")
             
-    
-    
+        # Date and Time Display
+        with gr.Row(elem_classes="datetime-display"):
+            date_display = gr.Textbox(label="Date", interactive=False)
+            time_display = gr.Textbox(label="Time", interactive=False)
         
-    # Title & Description
-    gr.Markdown("<h1 style='text-align: center; color: black;'>Equinix Chatbot for Automation Team</h1>")
+        # Add refresh button for time
+        refresh_btn = gr.Button("Update Date & Time")
+        refresh_btn.click(fn=update_datetime, inputs=[], outputs=[date_display, time_display])
         
-    # Date and Time Display
-    with gr.Row(elem_classes="datetime-display"):
-        date_display = gr.Textbox(label="Date", interactive=False)
-        time_display = gr.Textbox(label="Time", interactive=False)
-    
-    # Update date and time initially
-    date_val, time_val = get_current_datetime()
-    date_display.value = date_val
-    time_display.value = time_val
-    
-    # Add refresh button for time
-    refresh_btn = gr.Button("Update Date & Time")
-    refresh_btn.click(fn=update_datetime, inputs=[], outputs=[date_display, time_display])
-    
-    gr.Markdown("<p style='text-align: center; color: black;'>Ask me anything!</p>")
+        gr.Markdown("<p style='text-align: center; color: black;'>Ask me anything!</p>")
 
-    # Input & Dropdown Section
-    with gr.Row():
-        query_input = gr.Textbox(label="Enter your query")
-        query_method = gr.Dropdown(["Team Query", "General Query"], label="Select Query Type", value="Team Query")
+        # Input & Dropdown Section
+        with gr.Row():
+            query_input = gr.Textbox(label="Enter your query")
+            query_method = gr.Dropdown(["Team Query", "General Query"], label="Select Query Type", value="Team Query")
 
-    # Output Textbox
-    output_box = gr.Textbox(label="Response", interactive=False)
-    
-    # Buttons Section
-    with gr.Row():
-        submit_button = gr.Button("Submit")
-        reset_button = gr.Button("Reset Query")
-    
-    # Button Click Events
-    submit_button.click(
-        lambda query, method: query_router(query, method, retriever), 
-        inputs=[query_input, query_method], 
-        outputs=output_box
-    )
+        # Output Textbox
+        output_box = gr.Textbox(label="Response", interactive=False)
         
-    # Reset only the query input
-    reset_button.click(reset_query_field, inputs=[], outputs=[query_input,output_box])
-   
-
-    # Update date and time on submission
-    submit_button.click(
-        fn=update_datetime,
-        inputs=[],
-        outputs=[date_display, time_display]
-    )
+        # Buttons Section
+        with gr.Row():
+            submit_button = gr.Button("Submit")
+            reset_button = gr.Button("Reset Query")
+        
+        # Button Click Events
+        submit_button.click(
+            lambda query, method: query_router(query, method, retriever), 
+            inputs=[query_input, query_method], 
+            outputs=output_box
+        )
+            
+        # Reset only the query input
+        reset_button.click(reset_query_field, inputs=[], outputs=[query_input, output_box])
+       
+        # Update date and time on submission
+        submit_button.click(
+            fn=update_datetime,
+            inputs=[],
+            outputs=[date_display, time_display]
+        )
+        
+        # Initialize date and time values
+        date_val, time_val = get_current_datetime()
+        date_display.value = date_val
+        time_display.value = time_val
 
-    # Launch UI
-    ui.launch(share=True)    
+    # Launch the interface
+    demo.launch(share=True)
 
 if __name__ == "__main__":
-    main()
+    main()
+
+