added result multiple rows to show and csv file download result (half part)

app.py

@@ -9,7 +9,7 @@ import os
 import warnings
 
 
-from persistStorage import saveLog, getAllLogFilesPaths
+from persistStorage import saveLog, getAllLogFilesPaths, getNewCsvFilePath, removeAllCsvFiles
 from config import *
 from constants import *
 from utils import *
@@ -19,7 +19,7 @@ from queryHelperManagerCoT import QueryHelperChainOfThought
 
 
 pd.set_option('display.max_columns', None)
-pd.set_option('display.max_rows', 10)
+pd.set_option('display.max_rows', None)
 
 # Filter out all warning messages
 warnings.filterwarnings("ignore")
@@ -114,6 +114,33 @@ eg\n select * from schema.table limit 20\n\n"""
   sql = sqlparse.format(sql, reindent=True, keyword_case='upper')
   return sql, disclaimerOutputStripping
 
+def getResultFilePath():
+  global dbEngine, queryHelper
+  sql, disclaimerOutputStripping = preProcessSQL(sql=sql)
+  if not isDataQuery(sql):
+    return "Sorry not allowed to run. As the query modifies the data."
+  try:
+    dbEngine2 = DbEngine(dbCreds)
+    dbEngine2.connect()
+    conn = dbEngine2.getConnection()
+    df = pd.read_sql_query(sql, con=conn)
+    
+    dbEngine2.disconnect()
+    # return disclaimerOutputStripping + str(pd.DataFrame(df))
+  except Exception as e:
+    # errorMessage = {"function":"testSQL","error":str(e), "userInput":sql}
+    # saveLog(errorMessage, 'error')
+    dbEngine2.disconnect()
+    df = pd.DataFrame()
+    # print(f"Error occured during running the query {sql}.\n and the error is {str(e)}")
+
+  removeAllCsvFiles()
+  csvFilePath = getNewCsvFilePath()
+  df.to_csv(csvFilePath, index=False)
+  downloadableFilesPaths = getAllLogFilesPaths()
+  fileComponent = gr.File(csvFilePath)
+  return fileComponent
+
 def testSQL(sql):
   global dbEngine, queryHelper
 
@@ -235,6 +262,11 @@ with gr.Blocks() as demo:
         text_button = gr.Button("RUN QUERY")
         clear = gr.ClearButton([text_input, text_output])
         text_button.click(testSQL, inputs=text_input, outputs=text_output)
+
+        csvFileComponent = gr.File([], file_count='multiple')
+        downloadCsv = gr.Button("Get result as csv")
+        downloadCsv.click(getResultFilePath, inputs=text_input, outputs=csvFileComponent)
+        
      # screen 3 : To set creds, schema, tables and columns
     with gr.Tab("Setup"):
         gr.Markdown("""<h1><center> Run Query </center></h1>""")