added graph

app.py

@@ -1,6 +1,7 @@
 import yfinance as yf
 import pandas as pd
 import gradio as gr
+import plotly.graph_objs as go
 
 def fetch_and_rebalance(etfs_str, period, rebalance_threshold_percentage):
     etfs = [etf.strip() for etf in etfs_str.split(",")]
@@ -69,16 +70,23 @@ def fetch_and_rebalance(etfs_str, period, rebalance_threshold_percentage):
         return "No data was appended."
     final_value = df_rebalance["Total Value"].iloc[-1]
     total_return = (final_value - initial_investment) / initial_investment * 100
-    return f"Final Value: {final_value}, Total Return: {total_return}%"
+
+    # Create the plot
+    fig = go.Figure()
+    for etf in etfs:
+        fig.add_trace(go.Scatter(x=df_rebalance["Date"], y=df_rebalance[f"{etf} Value"], mode='lines', name=etf))
+    fig.update_layout(title='ETF Valuations Over Time', xaxis_title='Date', yaxis_title='Value')
+
+    return f"Final Value: {final_value}, Total Return: {total_return}%", fig
 
 interface = gr.Interface(
     fn=fetch_and_rebalance, 
     inputs=[
         gr.Textbox(lines=1, placeholder="ETFs (comma separated)", value="TQQQ,JEPI"), 
-        gr.Textbox(lines=1, placeholder="Period", value="1y"), 
-        gr.Slider(minimum=0.0, maximum=1.0, value=0.05, label="Rebalance Threshold Percentage")
+        gr.Textbox(lines=1, placeholder="Period", value="ytd"), 
+        gr.Slider(minimum=0.0, maximum=1.0, value=0.10, label="Rebalance Threshold Percentage")
     ],
-    outputs="text"
+    outputs=["text", "plot"]
 )
 
 interface.launch()

requirements.txt

@@ -3,3 +3,4 @@ pandas==1.3.3
 gradio==3.1
 httpx==0.23.0
 httpcore==0.15.0
+plotly==5.6.0