Ok

app.py

@@ -3,7 +3,6 @@ import requests
 import pandas as pd
 import plotly.express as px
 
-
 class DownloadsTracker:
     def __init__(self):
         self.all_packages = {}
@@ -27,8 +26,6 @@ class DownloadsTracker:
                     df["date"] = pd.to_datetime(df["date"])
                     df = df.sort_values("date")
                     df["cumulative_downloads"] = df["downloads"].cumsum()
-                    # Add weekly downloads
-                    df["weekly_downloads"] = df["downloads"].rolling(window=7).mean()
                     self.all_packages[pkg] = df
                 except:
                     errors.append(f"Error fetching {pkg}")
@@ -36,54 +33,25 @@ class DownloadsTracker:
             return "\n".join(errors)
         return None
 
+    def render(self, package_list, use_log_scale):
+        package_list = [p.strip() for p in package_list.split(",") if p.strip()]
+        self.current_packages = package_list
+        errors = self.fetch_if_needed(package_list)
+        return self.plot(use_log_scale), errors, gr.update(visible=errors is not None)
+    
     def plot(self, use_log_scale):
-        if not self.current_packages:
-            return px.line(title="No data to display")
-            
-        # Prepare data
-        df_list = []
+        fig = px.line(title="Cumulative Package Downloads")
         for pkg in self.current_packages:
             if pkg in self.all_packages:
-                df = self.all_packages[pkg].copy()
-                df['package'] = pkg
-                df_list.append(df)
-        
-        combined_df = pd.concat(df_list)
-        
-        # Create subplot figures
-        fig1 = px.line(combined_df, x="date", y="cumulative_downloads", color="package", title="Package Downloads")
-        fig2 = px.line(combined_df, x="date", y="weekly_downloads", color="package")
-        
-        # Combine into subplots
-        fig = px.line()
-        fig.add_traces(fig1.data)
-        fig.add_traces(fig2.data)
+                df = self.all_packages[pkg]
+                fig.add_scatter(x=df["date"], y=df["cumulative_downloads"], name=pkg)
         
-        # Update layout for subplots
-        fig.update_layout(
-            height=800,
-            grid={'rows': 2, 'columns': 1},
-            title_text="Package Downloads"
-        )
-        
-        # Move second set of traces to lower subplot
-        for i in range(len(fig2.data)):
-            fig.data[i + len(fig1.data)].update(yaxis='y2')
-        
-        # Update scales
         if use_log_scale:
             fig.update_yaxes(type="log")
         else:
             fig.update_yaxes(type="linear")
-        
         return fig
 
-    def render(self, package_list, use_log_scale):
-        package_list = [p.strip() for p in package_list.split(",") if p.strip()]
-        self.current_packages = package_list
-        errors = self.fetch_if_needed(package_list)
-        return self.plot(use_log_scale), errors, gr.update(visible=errors is not None)
-
 tracker = DownloadsTracker()
 
 css = """
@@ -91,15 +59,13 @@ css = """
 #textbox_id span {background-color: red}
 """
 
-with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     packages = gr.Textbox("transformers", label="Package names (comma-separated)")
-    log_scale = gr.Checkbox(label="Use logarithmic scale on Y-axis", value=False)
+    log_scale = gr.Checkbox(label="Use logarithmic scale", value=False)
     error_box = gr.Textbox(label="Errors:", interactive=False, visible=False, elem_id="textbox_id")
     render_btn = gr.Button("Render")
     plot = gr.Plot()
     render_btn.click(tracker.render, inputs=[packages, log_scale], outputs=[plot, error_box, error_box])
     log_scale.change(tracker.plot, inputs=[log_scale], outputs=[plot])
 
-    demo.load(tracker.render, inputs=[packages, log_scale], outputs=[plot, error_box, error_box])
-
-demo.launch()
+app.launch()