Update app.py

app.py

@@ -2,8 +2,15 @@ import gradio as gr
 import pandas as pd
 import plotly.express as px
 from transformers import pipeline
+from datasets import load_dataset
 
-# Sample data for CVEs
+# Load the additional datasets
+deepseek_prover_v1 = load_dataset('deepseek-ai/DeepSeek-Prover-V1', split='train')
+cybersecurity_kg = load_dataset('CyberPeace-Institute/Cybersecurity-Knowledge-Graph', split='train')
+codesearchnet_pep8 = load_dataset('kejian/codesearchnet-python-pep8-v1', split='train')
+code_text_python = load_dataset('semeru/code-text-python', split='train')
+
+# Sample CVE data (for visualization)
 cve_data = {
     'CVE ID': ['CVE-2023-0001', 'CVE-2023-0002', 'CVE-2023-0003', 'CVE-2023-0004', 'CVE-2023-0005'],
     'Severity': ['High', 'Medium', 'Low', 'High', 'Medium'],
@@ -17,7 +24,7 @@ cve_data = {
     'Published Date': ['2023-01-01', '2023-01-02', '2023-01-03', '2023-01-04', '2023-01-05']
 }
 
-# Convert the data to a DataFrame
+# Convert CVE data to a DataFrame
 cve_df = pd.DataFrame(cve_data)
 
 # Function to filter CVEs by severity
@@ -53,7 +60,6 @@ with gr.Blocks() as demo:
     # CVE Chart
     with gr.Row():
         cve_chart = gr.Plot(label='CVEs by Severity')
-        # Use update instead of plot
         cve_chart.update(generate_cve_chart())
 
     # Sentiment Analysis
@@ -65,5 +71,16 @@ with gr.Blocks() as demo:
     # Event listener for sentiment analysis
     analyze_btn.click(fn=analyze_sentiment, inputs=description_input, outputs=sentiment_output)
 
+    # Display additional datasets in the dashboard
+    with gr.Tab("Datasets Overview"):
+        gr.Markdown("## Overview of Additional Datasets")
+        
+        # Display datasets as dataframes
+        with gr.Row():
+            gr.Dataframe(label="DeepSeek-Prover-V1", value=deepseek_prover_v1)
+            gr.Dataframe(label="Cybersecurity Knowledge Graph", value=cybersecurity_kg)
+            gr.Dataframe(label="Code SearchNet Python PEP8", value=codesearchnet_pep8)
+            gr.Dataframe(label="Code Text Python", value=code_text_python)
+
 # Launch the app
-demo.launch(share=True)
+demo.launch(share=True)