Update app.py

app.py

@@ -1,19 +1,42 @@
+!pip install plotly
 import gradio as gr
 import plotly.graph_objects as go
 
-# Creating the figure
-fig = go.Figure()
-fig.add_trace(go.Bar(x=['Long Portfolio', 'Short Portfolio'], y=[90, 10], marker=dict(color=['blue', 'red'])))
+def plot_portfolio(long_percentage, short_percentage):
+    # Create a bar graph
+    fig = go.Figure()
 
+    # Add bars for long and short
+    fig.add_trace(go.Bar(x=['Long', 'Short'], y=[long_percentage, short_percentage],
+                        marker_color=['blue', 'red']))
+
+    # Update layout for better presentation
+    fig.update_layout(title_text='Portfolio Composition',
+                      xaxis_title='Portfolio',
+                      yaxis_title='Percentage',
+                      showlegend=False,  # Do not display legend
+                      width=500,         # Adjust the width
+                      height=400,        # Adjust the height
+                      margin=dict(l=50, r=50, b=50, t=50),  # Adjust margins
+                      )
+
+    # Save the graph to an image file
+    image_path = '/path/to/graph.png'
+    fig.write_image(image_path)
+
+    # Return the image path
+    return image_path
+
+# Define the Gradio interface
 iface = gr.Interface(
-    fn=lambda: fig.show(renderer="png", width=500, height=400),
-    live=False,
-    share=False,
-    theme="huggingface",  # You can adjust the theme if needed
+    fn=plot_portfolio,
+    inputs=[gr.Slider(0, 100, default=90, label='Long (%)'),
+            gr.Slider(0, 100, default=10, label='Short (%)')],
+    outputs=[gr.Image()],
+    live=True,
 )
 
-# Adjusting the style of the interface
-iface.layout(css=".output_container {width: 50%; margin: auto;}")  # Centering the graph
-
+# Launch the Gradio interface
 iface.launch()
 
+