Update app.py

app.py

@@ -10,6 +10,7 @@ from langchain.chains import LLMChain
 from datetime import datetime
 import pytz
 import time
+import shutil
 
 # Get API key from Hugging Face Spaces secrets
 google_api_key = os.environ.get("GOOGLE_API_KEY")
@@ -177,6 +178,19 @@ def check_logo_file(logo_path):
     print(f"Files in directory: {os.listdir()}")
     return os.path.exists(logo_path)
 
+# In your main function, before creating the UI:
+# Ensure the static directory exists
+os.makedirs("static", exist_ok=True)
+
+# Copy the logo file to the static directory (if it's not already there)
+src_logo = "Equinix-LOGO.jpeg"
+dst_logo = os.path.join("static", "Equinix-LOGO.jpeg")
+
+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
@@ -223,84 +237,80 @@ 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 Gradio UI
-    with gr.Blocks(css=custom_css) as ui:
-        # Check if logo exists before creating UI
-        logo_exists = check_logo_file(logo_path)
-        
-        # Header with logo
-        with gr.Row(elem_id="header-container"):
-            with gr.Column():
-                if logo_exists:
-                    # Use HTML for more control over logo display
-                    gr.HTML(f"""
-                        <div style="text-align:center;">
-                            <img id="logo-image" src="file/{logo_path}" alt="Equinix Logo" 
-                                 style="max-height:80px; margin:0 auto; display:block;">
-                        </div>
-                    """)
-                else:
-                    # Fallback text-based logo
-                    gr.HTML("""
-                        <div style="text-align:center; color:white; padding:10px;">
-                            <div style="color:#FF0000; font-size:24px;">■■■■</div>
-                            <div style="font-size:28px; letter-spacing:5px; margin-top:5px;">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)
-        
-        # 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])
+    # Title & Description
+    gr.Markdown("<h1 style='text-align: center; color: black;'>Equinix Chatbot for Automation Team</h1>")
         
-        gr.Markdown("<p style='text-align: center; color: black;'>Ask me anything!</p>")
+    # 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>")
 
-        # 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)
+    
+    # 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])
-       
+    # 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]
-        )
-    
-    # Launch UI
-    ui.launch(share=True)    
+    # Update date and time on submission
+    submit_button.click(
+        fn=update_datetime,
+        inputs=[],
+        outputs=[date_display, time_display]
+    )
+
+# Launch UI
+ui.launch(share=True)    
 
 if __name__ == "__main__":
     main()