Update lab/interim.py

lab/interim.py

@@ -55,7 +55,7 @@ def load_dataset_into_session():
         )
         if st.button("Load Hugging Face Dataset"):
             try:
-                dataset = load_dataset(dataset_name, split="train", trust_remote_code=True)
+                dataset = load_dataset(dataset_name, name="all", split="train", trust_remote_code=True, uniform_split=True)
                 if hasattr(dataset, "to_pandas"):
                     st.session_state.df = dataset.to_pandas()
                 else:
@@ -81,7 +81,7 @@ load_dataset_into_session()
 
 if "df" in st.session_state and llm:
     df = st.session_state.df
-    st.write("### Data Preview")
+    st.write("### Dataset Preview")
     st.dataframe(df.head(10))
 
     # Create SmartDataFrame
@@ -109,14 +109,14 @@ if "df" in st.session_state and llm:
                 # Save and display the plot
                 temp_plot_path = os.path.join(temp_dir, "plot.png")
                 plt.savefig(temp_plot_path)
-                st.image(temp_plot_path, caption="Generated Plot", use_column_width=True)
+                st.image(temp_plot_path, caption="Generated Plot", use_container_width=True)
 
         except Exception as e:
             st.error(f"Error: {e}")
 
 # Instructions
 with st.sidebar:
-    st.header("Instructions")
+    st.header("Instructions:")
     st.markdown(
         "1. Select how you want to input the dataset.\n"
         "2. Upload, select, or fetch the dataset using the provided options.\n"
@@ -127,8 +127,7 @@ with st.sidebar:
         "4. Enter a query to generate and view graphs based on patent attributes.\n"
     )
     st.markdown("---")
-    st.header("References")
+    st.header("References:")
     st.markdown(
         "1. [Chat With Your CSV File With PandasAI - Prince Krampah](https://medium.com/aimonks/chat-with-your-csv-file-with-pandasai-22232a13c7b7)"
-    )
-
+    )