Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 from huggingface_hub import hf_hub_download
 import importlib.util
+import base64
 
 # --- Private Repository Information ---
 PRIVATE_DATASET_ID = os.getenv("PRIVATE_DATASET_ID")
@@ -25,13 +26,21 @@ except Exception as e:
     def show_error(*args):
         return f"// ERROR: Failed to load core agent code. Details: {e}", "", ""
 
+# --- Base64 Encoding for Logo ---
+def get_logo_base64():
+    try:
+        with open("DeepV_logo.png", "rb") as image_file:
+            return base64.b64encode(image_file.read()).decode("utf-8")
+    except FileNotFoundError:
+        print("Logo file not found. Ensure DeepV_logo.png is in the correct directory.")
+        return ""
+
+logo_base64 = get_logo_base64()
+
 # --- Gradio UI setup below ---
 with gr.Blocks(title="DeepV for RTL (Model-Agnostic)", theme=gr.themes.Soft()) as demo:
     gr.Markdown("## DeepV for RTL Code Generation — Model-Agnostic (Bring Your Own API Key)", elem_id="main-title")
 
-    # The logo file component is now hidden
-    logo_file = gr.File("DeepV_logo.png", visible=False)
-    
     with gr.Row():
         with gr.Column(scale=2):
             with gr.Row():
@@ -74,11 +83,15 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)", theme=gr.themes.Soft()) a
                 )
             
         with gr.Column(scale=3):
-            # The logo is now an HTML component for maximum style control
-            # The HTML source is dynamically updated
-            logo_html = gr.HTML(
-                "<div id='deepv-logo-container'><img id='deepv-logo' src='' alt='DeepV Logo'></div>"
-            )
+            # The logo is now embedded directly in the HTML as a Base64 string
+            if logo_base64:
+                gr.HTML(f"""
+                    <div id="deepv-logo-container">
+                        <img id="deepv-logo" src="data:image/png;base64,{logo_base64}" alt="DeepV Logo">
+                    </div>
+                """)
+            else:
+                gr.Markdown("Logo not found.")
 
             out_code = gr.Textbox(
                 label="Generated Verilog",
@@ -95,19 +108,7 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)", theme=gr.themes.Soft()) a
                     value="Clear All",
                     components=[spec, out_code, temperature_tb, top_p_tb, max_new_tokens_tb]
                 )
-    
-    # --- Dynamic Logo Update ---
-    demo.load(
-        lambda path: gr.HTML(f"""
-            <div id='deepv-logo-container'>
-                <img id='deepv-logo' src='file={path}' alt='DeepV Logo'>
-            </div>
-        """),
-        inputs=logo_file,
-        outputs=logo_html,
-        show_progress=False
-    )
-    
+
     # --- Wrapper function to handle output from the generation agent ---
     def generate_only(
         spec, use_rag, top_k, model_choice, api_key, temperature, top_p, max_new_tokens