fixing faiss pointing

app.py

@@ -179,16 +179,20 @@ def ensure_index_downloaded() -> Path:
         local_dir=str(CACHE_DIR),
         local_dir_use_symlinks=False,
     )
+    
+    # --- START OF FIX ---
+    # Walk the downloaded directory to find the index files
+    downloaded_dir = Path(snapshot_path)
+    for root, dirs, files in os.walk(downloaded_dir):
+        if 'index.faiss' in files and 'index.pkl' in files:
+            index_path = Path(root)
+            print(f"[INFO] Found index files at {index_path}. Using this path.")
+            return index_path
 
-    # Prefer INDEX_SUBDIR if present
-    candidate = Path(snapshot_path) / INDEX_SUBDIR
-    if candidate.exists() and any(candidate.iterdir()):
-        print(f"[INFO] Found index files under {candidate}")
-        return candidate
+    # If the files are not found, raise an error
+    raise FileNotFoundError("FAISS index files (index.faiss and index.pkl) not found in the downloaded dataset.")
+    # --- END OF FIX ---
 
-    # Otherwise use snapshot root
-    print(f"[WARN] INDEX_SUBDIR='{INDEX_SUBDIR}' not found or empty; using snapshot root.")
-    return Path(snapshot_path)
 
 # Keep a lightweight global cache so we don’t reload embeddings on every click
 _VECTORSTORE_PATH = None
@@ -204,8 +208,8 @@ def run_generation(spec, use_rag, top_k, model_choice, api_key, temperature, top
         return "// Please provide a design specification and your API key.", "", []
 
     # Prepare agent
-    faiss_path = get_vectorstore_path()
     try:
+        faiss_path = get_vectorstore_path()
         agent = VerilogAgent(
             model_id=model_choice,
             embedding_id=EMBEDDING_MODEL,
@@ -262,8 +266,8 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
                         "gpt-4o",
                         "gpt-4o-mini",
                         "gpt-4.1",
-                        "gpt-5",        
-                        "gpt-5-mini"    
+                        "gpt-5",
+                        "gpt-5-mini"
                     ],
                     value="gpt-4o",
                     label="Model"
@@ -276,7 +280,6 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
                 max_new_tokens = gr.Slider(128, 4096, value=768, step=64, label="Max tokens")
 
             run_btn = gr.Button("Generate Verilog", variant="primary")
-# fixing
         with gr.Column(scale=3):
             gr.Markdown("**Output**")
             out_code = gr.Textbox(
@@ -301,4 +304,4 @@ with gr.Blocks(title="DeepV for RTL (Model-Agnostic)") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()