update app.py

app.py

@@ -1,7 +1,73 @@
 import gradio as gr
+import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
 
-def greet(name):
-    return "Hello " + name + "!!"
+from constants import *
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+
+from mpl_data_plotter import MatplotlibDataPlotter
+
+
+
+single_df = pd.read_csv(SINGLE_DOMAINS_FILE, compression='gzip')
+single_df['biosyn_class_index'] = single_df.bgc_class.apply(lambda x: BIOSYN_CLASS_NAMES.index(x))
+
+pair_df = pd.read_csv(PAIR_DOMAINS_FILE, compression='gzip')
+pair_df['biosyn_class_index'] = pair_df.bgc_class.apply(lambda x: BIOSYN_CLASS_NAMES.index(x))
+# unique_domain_lengths = single_df.dom_location_len.unique()
+num_domains_in_region_df = single_df.groupby('cds_region_id', as_index=False).agg({'as_domain_id': 'count'}).rename(
+    columns={'as_domain_id': 'num_domains'})
+
+unique_domain_lengths = num_domains_in_region_df.num_domains.unique()
+
+
+data_plotter = MatplotlibDataPlotter(single_df, pair_df, num_domains_in_region_df)
+
+
+def update_all_plots(frequency, split_name='stratified'):
+    return data_plotter.plot_single_domains(frequency, split_name), data_plotter.plot_pair_domains(frequency, split_name)
+
+
+# Create Gradio interface
+with gr.Blocks(title="Interactive Wave Plotter") as demo:
+    gr.Markdown("## Interactive Wave Plotter")
+    gr.Markdown("Adjust the slider to change the frequency of all waves simultaneously.")
+    
+    with gr.Row():
+        frequency_slider = gr.Slider(
+            minimum=unique_domain_lengths.min(), 
+            maximum=unique_domain_lengths.max(), 
+            step=1, 
+            value=unique_domain_lengths.min(), 
+            label="Min number of domains"
+        )
+    
+    with gr.Row():
+        with gr.Column():
+            single_domains_plot = gr.Plot(
+                label="Single domains",
+                container=True, 
+                elem_id="single_domains_plot"
+            )
+        # gr.HTML("""
+        # <style>
+        #     #single_domains_plot {
+        #         height: 100% !important;
+        #         width: 100% !important;
+        #     }
+        # </style>
+        # """)
+        with gr.Column():
+            pair_domains_plot = gr.Plot(label="Pair domains")
+        # with gr.Column():
+        #     combined_plot = gr.Plot(label="Combined Wave")
+    
+    frequency_slider.release(
+        fn=update_all_plots,
+        inputs=[frequency_slider],
+        outputs=[single_domains_plot, pair_domains_plot]#, cosine_plot]
+    )
+
+demo.launch()
+# demo.load(filter_map, [min_price, max_price, boroughs], map)