Update app.py

app.py

@@ -93,5 +93,30 @@ if __name__ == '__main__':
                 st.markdown("The SQL query is - " + db_chain.intermediate_steps['sql_cmd'])
             else:
                 st.markdown("Sorry I cannot answer that. Please try again later.")
+
+            feedback = st.radio("Does this answer your question?", ["Yes","No"])
+            if feedback == 'No':
+                with st.status("Retrieving results again..."):
+                    top_table_names = table_search(question, topk=1)['table'].tolist()
+                    print (top_table_names)
+                    try:
+                        db_chain._call(inputs={'query': question, 'history': []})
+                    except:
+                        pass
+                
+                if db_chain.intermediate_steps.get("sql_cmd_unchecked",'') == '':
+                    print (db_chain)
+                    st.markdown("Sorry I cannot answer that. Please try again later.")
+                elif db_chain.intermediate_steps.get("result",'') != '':
+                    st.markdown("Answer to your question is - " + db_chain.intermediate_steps.get("result",''))
+                    print (db_chain.intermediate_steps.get("result",''))
+                    st.markdown("The SQL query is - " + db_chain.intermediate_steps['sql_cmd'])
+                elif db_chain.intermediate_steps.get("sql_data",'') != '':
+                    out = pd.DataFrame.from_dict(db_chain.intermediate_steps['sql_data'])
+                    st.markdown("Here is your result in a table format")
+                    st.table(out)
+                    st.markdown("The SQL query is - " + db_chain.intermediate_steps['sql_cmd'])
+                else:
+                    st.markdown("Sorry I cannot answer that. Please try again later.")
         else:
             st.markdown("Your question is a generic one. This feature is under development. Please try again later.")