Upload app.py

gau from dropdown

app.py

@@ -10,6 +10,8 @@ import requests
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+LATEST_DF_BASE = None
+
 SAMPLE_RESPONSE: List[dict] = [
     {
         "название": "дрель-шуруповерт аккумуляторная 18 В, Li-Ion, быстрозажимной патрон 13 мм, 0-400/2000 об/мин, 91/58 Нм, 2-х скоростная, 21 уровень крутящего момента, встроенная подсветка, электрический тормоз, реверс, вес 2.3 кг, металлический редуктор, без аккумулятора и зарядного устройства  DHP458Z",
@@ -90,7 +92,9 @@ def _format_vertical(df: pd.DataFrame) -> pd.DataFrame:
     return vertical
 
 
-def extract_attributes(target_url: str) -> tuple[pd.DataFrame, str]:
+def extract_attributes(target_url: str):
+    global LATEST_DF_BASE
+
     if not target_url or not target_url.strip():
         raise gr.Error("Пожалуйста, введите URL для извлечения атрибутов.")
 
@@ -100,6 +104,42 @@ def extract_attributes(target_url: str) -> tuple[pd.DataFrame, str]:
 
     df = pd.DataFrame(data)
     df = df[_reorder_columns(df.columns)]
+
+    # Save base dataframe (without forced GAU) for future GAU updates
+    LATEST_DF_BASE = df.copy()
+
+    gau_choices: List[str] = []
+    selected_gau: str | None = None
+
+    if "ГАУ" in df.columns:
+        first_value = df["ГАУ"].iloc[0]
+        if isinstance(first_value, list):
+            gau_choices = [str(x).strip() for x in first_value if x]
+        elif pd.notna(first_value):
+            gau_choices = [str(first_value)]
+
+    if gau_choices:
+        selected_gau = gau_choices[0]
+        df["ГАУ"] = selected_gau
+
+    tsv_payload = _dataframe_to_tsv(df)
+    vertical_df = _format_vertical(df)
+
+    dropdown_update = gr.update(choices=gau_choices, value=selected_gau)
+    return vertical_df, tsv_payload, dropdown_update
+
+
+def update_gau(selected_gau: str):
+    global LATEST_DF_BASE
+
+    if LATEST_DF_BASE is None:
+        raise gr.Error("Сначала получите атрибуты по URL.")
+
+    df = LATEST_DF_BASE.copy()
+
+    if "ГАУ" in df.columns and selected_gau:
+        df["ГАУ"] = selected_gau
+
     tsv_payload = _dataframe_to_tsv(df)
     vertical_df = _format_vertical(df)
     return vertical_df, tsv_payload
@@ -149,6 +189,13 @@ def main() -> None:
             type="pandas",
             elem_id="results-table",
         )
+
+        gau_dropdown = gr.Dropdown(
+            label="Вероятные ГАУ (выберите наиболее подходящий)",
+            choices=[],
+            interactive=True,
+        )
+
         clipboard_box = gr.Textbox(
             label=None,
             interactive=False,
@@ -172,6 +219,12 @@ def main() -> None:
         submit_btn.click(
             fn=extract_attributes,
             inputs=url_input,
+            outputs=[results_table, clipboard_box, gau_dropdown],
+        )
+
+        gau_dropdown.change(
+            fn=update_gau,
+            inputs=gau_dropdown,
             outputs=[results_table, clipboard_box],
         )