handle candidate queries

app.py

@@ -99,45 +99,55 @@ if prompt:
         st.markdown(prompt)
 
     response = analyze_sales_data(sales_file, prompt)
+    print("==========>: ", response)
+
+    if response['status'] == 'success':
+        dummy_response = response['result'][0]
+
+        try:
+            # Process input
+            # values = list(map(int, prompt.strip().split(',')))
+            # labels = [f"Item {i+1}" for i in range(len(values))]
+            values = [item['y_axis'] for item in dummy_response['visualization']['data']]
+            labels = [item['x_axis'] for item in dummy_response['visualization']['data']]
+
+            # Create chart
+            fig, ax = plt.subplots()
+            if dummy_response['visualization']['type'] == 'line_chart':
+                ax.plot(labels, values, marker='o', color="lightcoral")
+            else:
+                ax.bar(labels, values, color="lightcoral")
+            plt.xticks(rotation=45, ha='right')
+            ax.set_title(dummy_response['visualization']['config']['title'])
+            ax.set_ylabel(dummy_response['visualization']['config']['y_axis'])
+            ax.set_xlabel(dummy_response['visualization']['config']['x_axis'])
+            
+            # Add value labels on top of each bar
+            for i, v in enumerate(values):
+                ax.text(i, v, f'{v:,.2f}', ha='center', va='bottom')
+
+            # Generate analysis message
+            analysis = "📊 Insights:\n"
+            for insight in dummy_response['insights']:
+                analysis += f"- {insight}\n"
+
+            # Add bot replies
+            st.session_state.messages.append({"role": "assistant", "content": fig})
+            st.session_state.messages.append({"role": "assistant", "content": analysis})
+            with st.chat_message("assistant"):
+                st.pyplot(fig)
+                st.markdown(analysis)
+
+        except Exception as e:
+            err_msg = f"❌ Error: {str(e)} — please enter only comma-separated numbers."
+            st.session_state.messages.append({"role": "assistant", "content": err_msg})
+            with st.chat_message("assistant"):
+                st.markdown(err_msg)
+    else:
+        err_msg = "❌ Error: Unable to process the request. Here are some example queries you can try:\n\n"
+        for candidate in response['candidate_questions']:
+            err_msg += f"- {candidate}\n"
 
-    dummy_response = response['result'][0]
-
-    try:
-        # Process input
-        # values = list(map(int, prompt.strip().split(',')))
-        # labels = [f"Item {i+1}" for i in range(len(values))]
-        values = [item['y_axis'] for item in dummy_response['visualization']['data']]
-        labels = [item['x_axis'] for item in dummy_response['visualization']['data']]
-
-        # Create chart
-        fig, ax = plt.subplots()
-        if dummy_response['visualization']['type'] == 'line_chart':
-            ax.plot(labels, values, marker='o', color="lightcoral")
-        else:
-            ax.bar(labels, values, color="lightcoral")
-        plt.xticks(rotation=45, ha='right')
-        ax.set_title(dummy_response['visualization']['config']['title'])
-        ax.set_ylabel(dummy_response['visualization']['config']['y_axis'])
-        ax.set_xlabel(dummy_response['visualization']['config']['x_axis'])
-        
-        # Add value labels on top of each bar
-        for i, v in enumerate(values):
-            ax.text(i, v, f'{v:,.2f}', ha='center', va='bottom')
-
-        # Generate analysis message
-        analysis = "📊 Insights:\n"
-        for insight in dummy_response['insights']:
-          analysis += f"- {insight}\n"
-
-        # Add bot replies
-        st.session_state.messages.append({"role": "assistant", "content": fig})
-        st.session_state.messages.append({"role": "assistant", "content": analysis})
-        with st.chat_message("assistant"):
-            st.pyplot(fig)
-            st.markdown(analysis)
-
-    except Exception as e:
-        err_msg = f"❌ Error: {str(e)} — please enter only comma-separated numbers."
         st.session_state.messages.append({"role": "assistant", "content": err_msg})
         with st.chat_message("assistant"):
             st.markdown(err_msg)