Update app.py

app.py

@@ -1,56 +1,41 @@
 import os
-from pathlib import Path
 import gradio as gr
-from huggingface_hub import snapshot_download, hf_hub_download
+from huggingface_hub import hf_hub_download
 import importlib.util
 
-# This is the path to your private dataset repository on Hugging Face Hub
+# --- Private Repository Information ---
 PRIVATE_DATASET_ID = os.getenv("PRIVATE_DATASET_ID")
 HF_TOKEN = os.getenv("HF_TOKEN")
 INDEX_SUBDIR = os.getenv("INDEX_SUBDIR", ".")
 
 # --- Core Logic Download and Import ---
 try:
-    # First, download the core agent code from the private repo
     AGENT_CODE_PATH = hf_hub_download(
         repo_id=PRIVATE_DATASET_ID,
-        filename="deepv_core.py",  # The file containing your VerilogAgent class
-        repo_type="dataset", # Ensure this is 'dataset' to match your repo type
+        filename="deepv_core.py",
+        repo_type="dataset",
         token=HF_TOKEN
     )
-    # Dynamically load the agent module from the downloaded file
     spec = importlib.util.spec_from_file_location("deepv_core_module", AGENT_CODE_PATH)
     agent_module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(agent_module)
 
-    # Now you can access the functions and classes from the private module
-    VerilogAgent = agent_module.VerilogAgent
+    # Now you can access the functions directly from the private module
     run_generation = agent_module.run_generation
-    get_vectorstore_path = agent_module.get_vectorstore_path
-    ensure_index_downloaded = agent_module.ensure_index_downloaded
     
 except Exception as e:
-    # Handle the error gracefully if the private repo can't be accessed
     def show_error(*args):
-        return f"// ERROR: Failed to load core agent code. Check your Hugging Face token and private dataset configuration. Details: {e}", "", []
-
-# Keep a lightweight global cache so we don’t reload embeddings on every click
-_VECTORSTORE_PATH = None
+        return f"// ERROR: Failed to load core agent code. Details: {e}", "", []
 
 # --- Gradio UI setup below ---
 with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
     gr.Markdown("## DeepV for RTL Code Generation — Model-Agnostic (Bring Your Own API Key)")
-
+    
     with gr.Row():
         with gr.Column(scale=2):
             with gr.Row():
                 model_choice = gr.Dropdown(
-                    choices=[
-                        "gpt-4o",
-                        "gpt-4o-mini",
-                        "gpt-4.1",
-                        "gpt-5-chat-latest",
-                    ],
+                    choices=["gpt-4o", "gpt-4o-mini", "gpt-4.1", "gpt-5-chat-latest"],
                     value="gpt-4o",
                     label="Model"
                 )
@@ -61,7 +46,7 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
                 **Note:** Your API key is used for the current session only and is not saved or stored.
                 """
             )
-
+            
             spec = gr.Textbox(
                 label="Design Specification (natural language or I/O contract)",
                 placeholder="e.g., 8-bit UART transmitter with baud rate generator ...",
@@ -70,7 +55,7 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
             with gr.Row():
                 use_rag = gr.Checkbox(value=True, label="Use Retrieval (RAG)")
                 top_k = gr.Slider(1, 10, value=3, step=1, label="Top-K retrieved examples")
-
+            
             with gr.Accordion("Generation Settings", open=False):
                 temperature = gr.Slider(0.0, 1.5, value=0.2, step=0.05, label="Temperature")
                 top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p")
@@ -85,7 +70,6 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
                 interactive=False,
                 placeholder="// Your Verilog code will appear here"
             )
-            # Add a dedicated copy button
             copy_button = gr.Button("Copy Code", variant="secondary")
 
             with gr.Tab("Retrieved Items (names + scores)"):
@@ -123,7 +107,6 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
         """
     )
 
-
 if __name__ == "__main__":
     if 'agent_module' in locals():
         demo.launch()