Update app.py (#13)

- Update app.py (473f974b0b8fbd33279ee6e1e3cc868bddaf8bbe)


Co-authored-by: Kendrick Manchester <kmanch3@users.noreply.huggingface.co>

app.py

@@ -55,6 +55,33 @@ CF_COL     = "Conductive Filler Conc. (wt%)"
 TARGET_COL = "Stress GF (MPa-1)"
 CANON_NA   = "NA"  # canonical placeholder for categoricals
 
+TYPE_CHOICES = [
+    "CNT",
+    "Brass fiber",
+    "GNP",
+    "Steel fiber",
+    "Carbon fiber",
+    "Graphene oxide",
+    "Graphene",
+    "Carbon black",
+    "Graphite",
+    "Shungite",
+    "Nickel powder",
+    "Glass cullet",
+    "MWCNT",
+    "Nano carbon black",
+    "Carbon powder",
+    "Gasification char",
+    "Used foundry sand",
+    "Nickel fiber",
+    "Nickel aggregate",
+    "Steel slag aggregate",
+    "TiO2",
+    "Carbonyl iron powder",
+    "Magnetite aggregate",
+    CANON_NA
+]
+
 MAIN_VARIABLES = [
     "Filler 1 Type",
     "Filler 1 Diameter (µm)",
@@ -671,14 +698,33 @@ def rag_reply(
     selected = mmr_select_sentences(question, hits, top_n=int(n_sentences), pool_per_chunk=6, lambda_div=0.7)
 
     # Header citations: short codes only, joined by '; ' (e.g., "S55; S71; S92")
-    header_codes = []
+    from urllib.parse import quote
+
+  
+
+    header_links = []
+    unique_codes = set()
+
     for _, r in hits.head(6).iterrows():
-        code = _short_doc_code(r["doc_path"])
-        if code not in header_codes:
-            header_codes.append(code)
-    header_cites = "; ".join(header_codes)
-    src_codes = set(header_codes)
-    coverage_note = "" if len(src_codes) >= 3 else f"\n\n> Note: Only {len(src_codes)} unique source(s) contributed. Add more PDFs or increase Top-K."
+        doc_path = r["doc_path"]
+        filename = Path(doc_path).name
+        short_code = _short_doc_code(doc_path)
+
+        # ✅ Correct Gradio route is /file= (NOT /file/)
+        abs_pdf = (LOCAL_PDF_DIR / filename).resolve()
+        href = f"/file={quote('papers/' + filename)}"
+        link = f'<a href="/file={quote("papers/" + filename)}" target="_blank" rel="noopener noreferrer">{short_code}</a>'
+
+        if short_code not in unique_codes:
+            header_links.append(link)
+            unique_codes.add(short_code)
+
+    header_cites = "; ".join(header_links)
+
+    coverage_note = "" if len(unique_codes) >= 3 else (
+    f"\n\n> Note: Only {len(unique_codes)} unique source(s) contributed. "
+    "Add more PDFs or increase Top-K."
+)
 
     # Prepare retrieval list for logging (full filenames kept here)
     retr_list = []
@@ -1026,6 +1072,41 @@ input[type="checkbox"], .gr-checkbox, .gr-checkbox > * { pointer-events: auto !i
 #eval-tab.eval-active .label {
   color: #e6f2ff !important;
 }
+
+/* --- THE UNIVERSAL DROPDOWN OVERRIDE --- */
+
+/* 1. All boxes show white text on the dark background */
+#filler-dropdown .single-select, #filler-dropdown input,
+#dim-dropdown .single-select, #dim-dropdown input,
+#dim2-dropdown .single-select, #dim2-dropdown input,
+#current-dropdown .single-select, #current-dropdown input {
+    color: #ffffff !important;
+    -webkit-text-fill-color: #ffffff !important;
+}
+
+/* 2. All dropdown menus (the pop-outs) have a white background */
+#filler-dropdown .options,
+#dim-dropdown .options,
+#dim2-dropdown .options,
+#current-dropdown .options {
+    background-color: #ffffff !important;
+}
+
+/* 3. All items in the lists are forced to PURE BLACK */
+#filler-dropdown .item, #filler-dropdown .item span,
+#dim-dropdown .item, #dim-dropdown .item span,
+#dim2-dropdown .item, #dim2-dropdown .item span,
+#current-dropdown .item, #current-dropdown .item span,
+.gr-dropdown .options .item, .gr-dropdown .options .item * {
+    color: #000000 !important;
+    -webkit-text-fill-color: #000000 !important;
+}
+
+/* 4. Hover effect for all dropdowns */
+.gr-dropdown .item:hover {
+    background-color: #dbeafe !important;
+}
+
 """
 
 theme = gr.themes.Soft(
@@ -1083,18 +1164,17 @@ with gr.Blocks(css=CSS, theme=theme, fill_height=True) as demo:
             with gr.Row():
                 with gr.Column(scale=7):
                     with gr.Accordion("Primary conductive filler", open=True, elem_classes=["card"]):
-                        f1_type = gr.Textbox(label="Filler 1 Type *", placeholder="e.g., CNT, Graphite, Steel fiber")
+                        f1_type = gr.Dropdown(TYPE_CHOICES,label="Filler 1 Type *", value="CNT", allow_custom_value=True, elem_id="filler-dropdown")
                         f1_diam = gr.Number(label="Filler 1 Diameter (µm) *")
                         f1_len  = gr.Number(label="Filler 1 Length (mm) *")
                         cf_conc = gr.Number(label=f"{CF_COL} *", info="Weight percent of total binder")
-                        f1_dim  = gr.Dropdown(DIM_CHOICES, value=CANON_NA, label="Filler 1 Dimensionality *")
+                        f1_dim  = gr.Dropdown(DIM_CHOICES, value=CANON_NA, label="Filler 1 Dimensionality *",elem_id="dim-dropdown")
 
                     with gr.Accordion("Secondary filler (optional)", open=False, elem_classes=["card"]):
                         f2_type = gr.Textbox(label="Filler 2 Type", placeholder="Optional")
                         f2_diam = gr.Number(label="Filler 2 Diameter (µm)")
                         f2_len  = gr.Number(label="Filler 2 Length (mm)")
-                        f2_dim  = gr.Dropdown(DIM_CHOICES, value=CANON_NA, label="Filler 2 Dimensionality")
-
+                        f2_dim  = gr.Dropdown(DIM_CHOICES, value=CANON_NA, label="Filler 2 Dimensionality", elem_id="dim2-dropdown")
                     with gr.Accordion("Mix design & specimen", open=False, elem_classes=["card"]):
                         spec_vol  = gr.Number(label="Specimen Volume (mm3) *")
                         probe_cnt = gr.Number(label="Probe Count *")
@@ -1112,7 +1192,7 @@ with gr.Blocks(css=CSS, theme=theme, fill_height=True) as demo:
                     with gr.Accordion("Mechanical & electrical loading", open=False, elem_classes=["card"]):
                         load_rate = gr.Number(label="Loading Rate (MPa/s)")
                         E_mod     = gr.Number(label="Modulus of Elasticity (GPa) *")
-                        current   = gr.Dropdown(CURRENT_CHOICES, value=CANON_NA, label="Current Type")
+                        current = gr.Dropdown(CURRENT_CHOICES, value=CANON_NA, label="Current Type", elem_id="current-dropdown")
                         voltage   = gr.Number(label="Applied Voltage (V)")
 
                 with gr.Column(scale=5):
@@ -1244,12 +1324,19 @@ with gr.Blocks(css=CSS, theme=theme, fill_height=True) as demo:
 
 # ------------- Launch -------------
 if __name__ == "__main__":
-    demo.queue().launch()
+    # 1. Start the Chatbot (This is what gives you the link)
+    # If using Gradio:
+    demo.launch() 
+    
+    # Or if using Flask:
+    # app.run(debug=True)
 
-    # After launch: export a simple list of PDFs as paper_list.csv
+    # 2. Everything below here only runs AFTER the server stops 
+    # (or might not run at all depending on how the server handles the exit)
     import os as _os
     import pandas as _pd
     folder = "papers"
     files = sorted(_os.listdir(folder)) if _os.path.exists(folder) else []
     _pd.DataFrame({"doc": files}).to_csv("paper_list.csv", index=False)
     print("✅ Saved paper_list.csv with", len(files), "papers")
+