model change

alpha_chart.py

@@ -0,0 +1,35 @@
+import matplotlib.pyplot as plt
+import requests
+import os
+
+def plot_alpha_btc_chart_to_file(portfolio_id: str, output_path="alpha_chart.png"):
+    url = f"https://api.tradelink.pro/portfolio/get?portfolioId={portfolio_id}&extended=1&declaration=1&step=day&lang=en&incViews=1"
+    headers = {"User-Agent": "Mozilla/5.0", "Accept": "application/json"}
+
+    try:
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        alpha_data = response.json()["data"]["extended"]["alphaBTC"]
+    except Exception as e:
+        print(f"[Ошибка]: {e}")
+        return None
+
+    values = [item["value"] for item in alpha_data]
+    indices = list(range(len(values)))
+
+    fig, ax = plt.subplots(figsize=(12, 6))
+    ax.plot(indices, values, color='blue', linewidth=1.5, label='alphaBTC')
+    ax.axhline(0, color='black', linewidth=1)
+    ax.spines['right'].set_color('none')
+    ax.spines['top'].set_color('none')
+    ax.spines['bottom'].set_position('zero')
+    ax.spines['left'].set_position('zero')
+    ax.set_title("Alpha к BTC", fontsize=14)
+    ax.set_xlabel("Индекс")
+    ax.set_ylabel("Alpha")
+    ax.grid(True, linestyle='--', alpha=0.5)
+    ax.legend()
+    plt.tight_layout()
+    plt.savefig(output_path)
+    plt.close()
+    return output_path

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 import os
+import asyncio
 from prompts import TRADELINK_CONTEXT
 from openai import OpenAI
 from analyzer import analyze_portfolio_streaming
 from comparer import compare_portfolio
 from table_view import get_metrics_dataframe
-import asyncio
+from alpha_chart import plot_alpha_btc_chart_to_file
+
 
 
 api_key = os.getenv("featherless")  #
@@ -82,5 +84,18 @@ with gr.Blocks() as demo:
 
         metrics_button.click(fn=show_metrics_table, inputs=metrics_input, outputs=metrics_output)
 
+
+
+    with gr.Tab("📈 AlphaBTC график"):
+        chart_input = gr.Textbox(label="Введите portfolioId")
+        chart_button = gr.Button("Построить график")
+        chart_image = gr.Image(label="Alpha график")
+
+        def get_chart_image(portfolio_id):
+            path = plot_alpha_btc_chart_to_file(portfolio_id)
+            return path if path else None
+
+        chart_button.click(fn=get_chart_image, inputs=chart_input, outputs=chart_image)
+
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -2,4 +2,5 @@ gradio>=4.29.0
 openai>=1.30.1
 requests
 httpx
-pandas
+pandas
+matplotlib