test5

app.py

@@ -9,7 +9,7 @@ client = OpenAI(
     api_key="rc_2b641fdc25954733a5ef7ccd10ce6a8e1db603e386a0fca262626daf21562b4b"
 )
 
-# Функция извлечения UUID
+# Извлечение UUID из строки
 def extract_portfolio_id(text: str) -> str | None:
     match = re.search(
         r"[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}",
@@ -17,69 +17,118 @@ def extract_portfolio_id(text: str) -> str | None:
     )
     return match.group(0) if match else None
 
-# Генератор — стриминговый вывод анализа
+# Получение метрик
+def fetch_metrics(portfolio_id: str) -> dict | None:
+    try:
+        url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1&declaration=1&step=day&lang=en&incViews=1"
+        response = requests.get(url)
+        response.raise_for_status()
+        extended = response.json().get("data", {}).get("extended", {})
+        keys = [
+            # 🔹 Доходность и риск
+            "alphaRatio", "betaRatio", "cagr", "sharpe", "sortino", "volatility",
+            "kSortino", "kCalmar", "kSharpe",
+
+            # 🔹 Просадки
+            "maxDD", "mddDuration", "maxBalance",
+
+            # 🔹 Поведение стратегии
+            "losingDays", "winningDays", "selfProfitRate",
+
+            # 🔹 Актуальные показатели доходности
+            "lastWeekNetProfit", "lastMonthNetProfit", "lastQuarterGrowth", "lastYearNetGrowth"
+        ]
+
+        return {k: extended[k] for k in keys if isinstance(extended.get(k), (int, float))}
+    except:
+        return None
+
+# Анализ одного портфеля
 def analyze_portfolio_streaming(text: str):
     portfolio_id = extract_portfolio_id(text)
     if not portfolio_id:
-        yield "❗ Укажите корректный portfolioId или ссылку на портфель."
+        yield "❗ Укажите корректный portfolioId или ссылку."
         return
 
-    url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1&declaration=1&step=day&lang=en&incViews=1"
+    metrics = fetch_metrics(portfolio_id)
+    if not metrics:
+        yield "❗ Не удалось получить метрики портфеля."
+        return
 
-    try:
-        response = requests.get(url)
-        response.raise_for_status()
-        json_data = response.json()
-        extended = json_data.get("data", {}).get("extended", {})
-
-        metrics_keys = ["alphaRatio", "betaRatio", "cagr", "sharpe", "sortino", "volatility"]
-        filtered = {k: v for k, v in extended.items() if k in metrics_keys and isinstance(v, (int, float))}
-
-        if not filtered:
-            yield "❗ Ключевые метрики отсутствуют в ответе API."
-            return
-
-        metrics_text = ", ".join([f"{k}: {v}" for k, v in filtered.items()])
-        prompt = f"""Вот метрики портфеля: {metrics_text}.
-Проанализируй их и объясни сильные и слабые стороны на русском языке, как финансовый аналитик."""
-
-        response_llm = client.chat.completions.create(
-            model="meta-llama/Meta-Llama-3.1-8B-Instruct",
-            messages=[
-                {"role": "system", "content": "Ты — финансовый аналитик."},
-                {"role": "user", "content": prompt}
-            ],
-            stream=True
-        )
-
-        partial = ""
-        for chunk in response_llm:
-            delta = chunk.choices[0].delta.content
-            if delta:
-                partial += delta
-                yield partial
-
-    except Exception as e:
-        yield f"❌ Ошибка: {e}"
-
-# Интерфейс Gradio (Blocks)
-with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Анализ инвестиционного портфеля Tradelink")
+    metrics_text = ", ".join([f"{k}: {v}" for k, v in metrics.items()])
+    prompt = f"Вот метрики портфеля: {metrics_text}. Проанализируй их и объясни сильные и слабые стороны на русском языке, как финансовый аналитик."
 
-    with gr.Row():
-        portfolio_input = gr.Textbox(
-            label="Введите ссылку или portfolioId",
-            placeholder="ea856c1b-65ee-4177-b8d1-e5ab0d610a91"
-        )
-        analyze_button = gr.Button("🔍 Проанализировать")
+    response_llm = client.chat.completions.create(
+        model="meta-llama/Meta-Llama-3.1-8B-Instruct",
+        messages=[
+            {"role": "system", "content": "Ты — финансовый аналитик."},
+            {"role": "user", "content": prompt}
+        ],
+        stream=True
+    )
+
+    partial = ""
+    for chunk in response_llm:
+        delta = chunk.choices[0].delta.content
+        if delta:
+            partial += delta
+            yield partial
+
+# Сравнение двух портфелей
+def compare_portfolios_streaming(text1: str, text2: str):
+    id1 = extract_portfolio_id(text1)
+    id2 = extract_portfolio_id(text2)
+    if not id1 or not id2:
+        yield "❗ Один или оба portfolioId некорректны."
+        return
+
+    m1 = fetch_metrics(id1)
+    m2 = fetch_metrics(id2)
+    if not m1 or not m2:
+        yield "❗ Не удалось получить метрики одного из портфелей."
+        return
+
+    m1_text = ", ".join([f"{k}: {v}" for k, v in m1.items()])
+    m2_text = ", ".join([f"{k}: {v}" for k, v in m2.items()])
+    prompt = f"""Сравни два инвестиционных портфеля:
 
-    output_box = gr.Textbox(label="📈 Результат анализа", lines=15)
+Портфель 1: {m1_text}
+Портфель 2: {m2_text}
 
-    analyze_button.click(
-        fn=analyze_portfolio_streaming,
-        inputs=portfolio_input,
-        outputs=output_box
+Сделай сравнительный анализ на русском языке как финансовый аналитик. Укажи, какой портфель сильнее, в чём риски, где преимущества."""
+
+    response = client.chat.completions.create(
+        model="meta-llama/Meta-Llama-3.1-8B-Instruct",
+        messages=[
+            {"role": "system", "content": "Ты — финансовый аналитик."},
+            {"role": "user", "content": prompt}
+        ],
+        stream=True
     )
 
+    partial = ""
+    for chunk in response:
+        delta = chunk.choices[0].delta.content
+        if delta:
+            partial += delta
+            yield partial
+
+# Gradio интерфейс
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 Анализ и сравнение инвестиционных портфелей Tradelink")
+
+    with gr.Tab("📊 Анализ"):
+        portfolio_input = gr.Textbox(label="Введите ссылку или portfolioId", placeholder="ea856c1b-...")
+        analyze_button = gr.Button("🔍 Проанализировать")
+        output_box = gr.Textbox(label="📈 Результат анализа", lines=15)
+        analyze_button.click(fn=analyze_portfolio_streaming, inputs=portfolio_input, outputs=output_box)
+
+    with gr.Tab("⚖️ Сравнение"):
+        compare_input_1 = gr.Textbox(label="Портфель 1", placeholder="ea856c1b-...")
+        compare_input_2 = gr.Textbox(label="Портфель 2", placeholder="d52f55cc-...")
+        compare_button = gr.Button("📊 Сравнить")
+        compare_output = gr.Textbox(label="📉 Результат сравнения", lines=20)
+        compare_button.click(fn=compare_portfolios_streaming, inputs=[compare_input_1, compare_input_2], outputs=compare_output)
+
 if __name__ == "__main__":
     demo.launch()