gemini recreate merge balance tool

README.md

@@ -11,3 +11,29 @@ short_description: Набор инструментов для баланса м
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+Задача сделать тулзу для удобной работы с балансом мердж-2 игры.
+
+1) Одна из основных фичей тулзы это Симуляция работы Генератора Заказов на базе конфигов с его правилами и конфигов мердж-цепочек. На выходе мы получаем таблицу с данными по симуляции и небольшой текстовый файл с сводными данными.
+
+2) В этой тулзе также можно менять параметры разных мердж конфигов - опять же если нужно протестировать значения. Значит важно чтобы мы могли загружать оригинальные файлы конфигов (.asset) и в интерфейсе удобно автоматически раскладывать их по типу и сразу исключая лишние данные (например ключи переводов и тд) - оставлять только важные параметры влияющие на баланс игры.
+
+3) Расчет стоимости предметов мердж цепочки исходя из весов предметов требуемых для производства других уровней предметов цепочки выпадающих из спавнеров.
+
+
+
+MergeOrderSimulationService - проверенный, рабочий в юнити симулятор генератора заказов который нам нужно переписать под новый инструмент.
+
+MergeChainCostCalculator - инструмент для расчета стоимости в энергии предметов мердж цепочек. Скрипт пока не работает должным образом. Хорошо если мы его починим для нашего инструмента.
+
+
+
+Далее.
+
+Наша задача взять все эти данные, сделать анализ конфигов для импорта, забрать наработки, сделать удобный интерфейс и произвести стабильный расширяемый при необходимости инструмент для балансировки игровых механик.
+
+
+
+Я хочу, чтобы инструмент был сделан реализован на библиотеке и компонентах gradio.app на python.
+

app.py

@@ -1,7 +1,260 @@
 import gradio as gr
+import pandas as pd
+import yaml
+import io
+import random
+from collections import deque
 
-def greet(name):
-    return "Hello " + name + "!!"
+# HELPER FUNCTIONS TO PARSE .asset FILES
+def parse_asset_file(file):
+    """Parses a YAML-like .asset file, skipping the initial %YAML and %TAG lines."""
+    if file is None:
+        return None
+    try:
+        content = file.decode('utf-8')
+        # Find the first '---' which indicates the start of the YAML document
+        yaml_start = content.find('---')
+        if yaml_start == -1:
+            raise ValueError("Not a valid Unity .asset file (missing '---')")
+        
+        # Clean up the content for standard YAML parsing
+        yaml_content = content[yaml_start:]
+        # Remove the MonoBehaviour header which is not valid YAML
+        lines = yaml_content.split('\n')
+        cleaned_lines = [line for line in lines if not line.strip().startswith('MonoBehaviour:')]
+        cleaned_yaml = '\n'.join(cleaned_lines)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+        # The file seems to have a custom serialization format within 'serializationData'.
+        # For now, we will parse the main structure.
+        # A more robust solution might need a dedicated parser for Unity's format.
+        
+        # Let's try to parse the main body first
+        data = yaml.safe_load(io.StringIO(cleaned_yaml))
+        
+        # A simplified approach to extract key-value pairs for now
+        # This will need to be much more robust based on the actual file structure.
+        config_data = {}
+        for line in content.split('\n'):
+            if ':' in line:
+                parts = line.split(':', 1)
+                key = parts[0].strip()
+                value = parts[1].strip()
+                if key.startswith('_'):
+                    config_data[key] = value
+
+        return config_data, data
+    except Exception as e:
+        print(f"Error parsing file: {e}")
+        return None, None
+
+# --- SIMULATION LOGIC ---
+def run_simulation_logic(ruleset_file, chain_files, iteration_count, initial_energy):
+    """
+    This function will contain the Python port of the C# MergeOrderSimulationService.
+    For now, it returns placeholder data.
+    """
+    # Placeholder for parsed ruleset
+    ruleset_id = "ruleset_placeholder"
+    
+    # Placeholder for selected chain IDs
+    selected_chain_ids = [f"chain_{i}" for i in range(len(chain_files))]
+    
+    # --- This is where the core simulation logic will go ---
+    # We will replicate the steps from MergeOrderSimulationService.cs:
+    # 1. Get all available items from selected chains.
+    # 2. Apply MergeBoardItemFinder simulation.
+    # 3. Filter by unlock level.
+    # 4. Filter recently used items.
+    # 5. Generate order with proper difficulty.
+    # 6. Update history.
+    # 7. Calculate costs and rewards.
+    # 8. Store data for CSV.
+    
+    # For now, let's generate some dummy data to show in the UI
+    sim_data = []
+    current_energy = initial_energy
+    difficulty_history = deque(maxlen=5) # Assuming max history of 5 from ruleset
+
+    for i in range(iteration_count):
+        req1 = f"merge_item_wood_branch_{random.randint(1, 3)}"
+        req2 = f"merge_item_stones_stones_{random.randint(1, 2)}" if random.random() > 0.5 else ""
+        energy_cost = random.randint(5, 20)
+        current_energy -= energy_cost
+        total_difficulty = random.randint(500, 2000)
+        difficulty_history.append(total_difficulty)
+        
+        sim_data.append({
+            "Order": i + 1,
+            "MEnergy_Price": energy_cost,
+            "MEnergy_Amount": current_energy,
+            "Requirement_1": req1,
+            "Weight_1": random.randint(50, 150),
+            "ChainId_1": "orders_merge_wood",
+            "Level_1": int(req1.split('_')[-1]),
+            "Requirement_2": req2,
+            "Weight_2": random.randint(75, 125) if req2 else 0,
+            "ChainId_2": "merge_stones" if req2 else "",
+            "Level_2": int(req2.split('_')[-1]) if req2 else 0,
+            "Total_Difficulty": total_difficulty,
+            "Adjusted_Difficulty": total_difficulty + sum(list(difficulty_history)[:-1]),
+        })
+
+    df = pd.DataFrame(sim_data)
+    
+    # Placeholder for statistics report
+    stats_report = f"""
+SIMULATION STATISTICS
+=====================
+Total Orders Generated: {iteration_count}
+Average Difficulty: {df['Total_Difficulty'].mean():.2f}
+
+Item Frequency (Top 5):
+{df['Requirement_1'].value_counts().head(5).to_string()}
+    """
+    
+    return df, stats_report, "simulation_results.csv", "simulation_stats.txt"
+
+def run_simulation_interface(ruleset_file, chain_files, iteration_count, initial_energy):
+    if not ruleset_file or not chain_files:
+        raise gr.Error("Пожалуйста, загрузите файл правил и хотя бы один файл цепочки.")
+    
+    df, stats_report, df_path, report_path = run_simulation_logic(
+        ruleset_file, chain_files, iteration_count, initial_energy
+    )
+    
+    # Convert DataFrame to a downloadable CSV file
+    csv_buffer = io.StringIO()
+    df.to_csv(csv_buffer, index=False)
+    csv_bytes = csv_buffer.getvalue().encode('utf-8')
+
+    return df, stats_report, gr.File.update(value=csv_bytes, visible=True, name=df_path), gr.File.update(value=stats_report.encode('utf-8'), visible=True, name=report_path)
+
+# --- COST CALCULATION LOGIC ---
+def run_cost_calculation_logic(chain_files, item_files):
+    """
+    This function will contain the Python port of the C# MergeChainCostCalculator.
+    For now, it returns placeholder data.
+    """
+    # --- This is where the core cost calculation logic will go ---
+    # We will replicate the steps from MergeChainCostCalculator.cs:
+    # 1. Load and parse all merge chain and item configs.
+    # 2. For each chain, find its associated producer (spawner).
+    # 3. For each item in the chain, calculate the expected number of attempts to get it.
+    # 4. Calculate the final energy cost (expected attempts * spawn cost).
+    # 5. Handle dynamic weights (IncWeight).
+    
+    # Dummy data for now
+    cost_data = []
+    for i, chain_file in enumerate(chain_files):
+        chain_id = f"chain_{i+1}"
+        for level in range(1, 7):
+            cost_data.append({
+                "Chain ID": chain_id,
+                "Item ID": f"{chain_id}_item_{level}",
+                "Level": level,
+                "Producer ID": f"spawner_{i+1}",
+                "Spawn Cost": 5,
+                "Weight": 100 - (level * 10),
+                "Inc Weight": 0,
+                "Expected Attempts": 1.0 + (level * 0.5),
+                "Energy Cost": (1.0 + (level * 0.5)) * 5,
+                "Warning": ""
+            })
+            
+    df = pd.DataFrame(cost_data)
+    return df
+
+def run_cost_calculation_interface(chain_files, item_files):
+    if not chain_files or not item_files:
+        raise gr.Error("Пожалуйста, загрузите файлы цепочек и предметов.")
+        
+    df = run_cost_calculation_logic(chain_files, item_files)
+    return df
+
+# --- CONFIG EDITOR LOGIC ---
+def view_config_file(file):
+    if file is None:
+        return "Загрузите файл для просмотра.", ""
+    
+    raw_content = file.decode('utf-8')
+    
+    # A simple display of the raw content.
+    # In a real scenario, we would parse this into editable components.
+    return raw_content, os.path.basename(file.name)
+
+# --- GRADIO UI ---
+with gr.Blocks(theme=gr.themes.Soft(), title="Инструмент Балансировки Merge-2") as demo:
+    gr.Markdown("# Инструмент для Балансировки Игр Merge-2")
+
+    with gr.Tabs():
+        # --- SIMULATION TAB ---
+        with gr.TabItem("Симуляция генератора заказов"):
+            gr.Markdown("## Симуляция генератора заказов\nЗагрузите конфиги и запустите симуляцию для получения данных по генерации заказов.")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    sim_ruleset_upload = gr.File(label="Загрузить Ruleset (.asset)")
+                    sim_chains_upload = gr.File(label="Загрузить файлы цепочек (.asset)", file_count="multiple")
+                    sim_iterations = gr.Slider(10, 1000, value=100, step=10, label="Количество итераций")
+                    sim_initial_energy = gr.Number(value=10000, label="Начальное количество энергии")
+                    sim_run_button = gr.Button("Запустить симуляцию", variant="primary")
+                with gr.Column(scale=3):
+                    gr.Markdown("### Результаты симуляции")
+                    sim_results_df = gr.DataFrame(label="Данные по заказам")
+                    gr.Markdown("### Сводный отчет")
+                    sim_stats_report = gr.Textbox(label="Статистика", lines=10)
+                    with gr.Row():
+                         sim_download_csv = gr.File(label="Скачать CSV", visible=False)
+                         sim_download_report = gr.File(label="Скачать отчет", visible=False)
+
+        # --- COST CALCULATION TAB ---
+        with gr.TabItem("Расчет стоимости цепочек"):
+            gr.Markdown("## Расчет стоимости предметов\nЗагрузите конфиги цепочек и спавнеров для расчета стоимости производства предметов в энергии.")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    cost_chains_upload = gr.File(label="Загрузить файлы цепочек (.asset)", file_count="multiple")
+                    cost_items_upload = gr.File(label="Загрузить файлы спавнеров/предметов (.asset)", file_count="multiple")
+                    cost_run_button = gr.Button("Рассчитать стоимость", variant="primary")
+                with gr.Column(scale=3):
+                    gr.Markdown("### Рассчитанная стоимость")
+                    cost_results_df = gr.DataFrame(label="Стоимость предметов в энергии")
+
+        # --- CONFIG EDITOR TAB ---
+        with gr.TabItem("Редактор конфигов"):
+            gr.Markdown("## Редактор Конфигурационных Файлов\nЗагрузите файл для просмотра и редактирования его параметров.")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    config_upload = gr.File(label="Загрузить файл .asset для редактирования")
+                    config_file_name = gr.Label(label="Выбранный файл")
+                with gr.Column(scale=2):
+                    # For now, just a textbox. Could be replaced with dynamic fields.
+                    config_content_viewer = gr.Textbox(label="Содержимое файла", lines=30, interactive=True)
+                    config_save_button = gr.Button("Сохранить изменения (не реализовано)", interactive=False)
+    
+    # --- Event Handlers ---
+    sim_run_button.click(
+        fn=run_simulation_interface,
+        inputs=[sim_ruleset_upload, sim_chains_upload, sim_iterations, sim_initial_energy],
+        outputs=[sim_results_df, sim_stats_report, sim_download_csv, sim_download_report]
+    )
+    
+    cost_run_button.click(
+        fn=run_cost_calculation_interface,
+        inputs=[cost_chains_upload, cost_items_upload],
+        outputs=[cost_results_df]
+    )
+
+    config_upload.upload(
+        fn=lambda x: x.name if x else "",
+        inputs=config_upload,
+        outputs=config_file_name
+    )
+    
+    config_upload.upload(
+        fn=lambda file: file.decode('utf-8') if file else "",
+        inputs=config_upload,
+        outputs=config_content_viewer
+    )
+
+
+if __name__ == "__main__":
+    demo.launch()