Update lab/interim.py

lab/interim.py

@@ -128,8 +128,8 @@ if "df" in st.session_state and llm:
     chat_df = SmartDataframe(df, config={"llm": llm})
 
     # Chat functionality
-    st.write("### Chat with Your Patent Data")
-    user_query = st.text_input("Enter your question about the patent data (e.g., 'Predict if the patent will be accepted.'):")
+    st.write("### Chat with Patent Data")
+    user_query = st.text_input("Enter your question about the patent data:", value = "Have the patents with the numbers 14908945, 14994130, 14909084, and 14995057 been accepted or rejected? What are their titles?")
 
     if user_query:
         try:
@@ -140,7 +140,7 @@ if "df" in st.session_state and llm:
 
     # Plot generation functionality
     st.write("### Generate and View Graphs")
-    plot_query = st.text_input("Enter a query to generate a graph (e.g., 'Plot the number of patents by filing year.'):")
+    plot_query = st.text_input("Enter a query to generate a graph:", value = "What is the distribution of patents categorized as 'ACCEPTED', 'REJECTED', or 'PENDING'?")
 
     if plot_query:
         try:
@@ -157,17 +157,17 @@ if "df" in st.session_state and llm:
             st.error(f"Error: {e}")
 
     # Download processed dataset
-    st.write("### Download Processed Dataset")
-    st.download_button(
-        label="Download Dataset as CSV",
-        data=df.to_csv(index=False),
-        file_name="processed_dataset.csv",
-        mime="text/csv"
-    )
+    #st.write("### Download Processed Dataset")
+    #st.download_button(
+     #   label="Download Dataset as CSV",
+     #   data=df.to_csv(index=False),
+     #   file_name="processed_dataset.csv",
+     #   mime="text/csv"
+    #)
 
 # Sidebar instructions
 with st.sidebar:
-    st.header("Instructions:")
+    st.header("📋 Instructions:")
     st.markdown(
         "1. Choose an LLM (Groq-based or OpenAI-based) to interact with the data.\n"
         "2. Upload, select, or fetch the dataset using the provided options.\n"
@@ -177,7 +177,7 @@ with st.sidebar:
         "   - Example: 'Summarize the abstract of this patent.'\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)"
-    )
+    )