Integrate changes for using OPen AI (#2)

Co-authored-by: adityatomar <[email protected]>

Author: tomaad

apis.py

@@ -1,6 +1,6 @@
 import os
 import kustoQuery
-import utils
+import apis
 from azure.kusto.data import KustoClient
 from azure.kusto.data.exceptions import KustoServiceError
 from azure.kusto.data.helpers import dataframe_from_result_table
@@ -21,35 +21,37 @@ def favicon():
     return send_from_directory(os.path.join(app.root_path, 'static'),
                                'favicon.ico', mimetype='image/vnd.microsoft.icon')
 
-@app.route('/hello', methods=['POST'])
-def hello():
-   name = request.form.get('name')
+@app.route('/execute_prompt', methods=['POST'])
+def execute_prompt():
+   user_input = request.form.get('prompt')
 
-   if name:
-        print('Request for hello page received with name=%s' % name)
-
-        print("Kusto Query App is starting...")
+   if user_input:
+        print('Prompt=%s' % user_input)
+        response = apis.call_openai(apis.system_prompt, user_input)
+        # Printing the result
+        print(response)
+        # print("Kusto Query App is starting...")
 
         app = kustoQuery.KustoQueryApp()
         app.load_configs(app.CONFIG_FILE_NAME)
 
-        if app.config.authentication_mode == "UserPrompt":
-            app.wait_for_user_to_proceed("You will be prompted for credentials during this script. Please return to the console after authenticating.")
+        # if app.config.authentication_mode == "UserPrompt":
+        #     app.wait_for_user_to_proceed("You will be prompted for credentials during this script. Please return to the console after authenticating.")
 
-        kusto_connection_string = utils.Utils.Authentication.generate_connection_string(app.config.kusto_uri, app.config.authentication_mode)
+        kusto_connection_string = apis.Utils.Authentication.generate_connection_string(app.config.kusto_uri, app.config.authentication_mode)
         print(f"Using cluster URI: {app.config.kusto_uri}")
 
         if not kusto_connection_string:
-            utils.Utils.error_handler("Connection String error. Please validate your configuration file.")
+            apis.Utils.error_handler("Connection String error. Please validate your configuration file.")
         else:
             with KustoClient(kusto_connection_string) as kusto_client:
-                app.query_table(kusto_client, app.config.database_name, app.config.table_name)
+                df = app.query_table(kusto_client, app.config.database_name, app.config.table_name, response)
 
         print("\nKusto Query App done")
 
-        return render_template('hello.html', name = name)
+        return render_template('response.html', output = df.to_string(index=False))
    else:
-        print('Request for hello page received with no name or blank name -- redirecting')
+        print('Request received without any prompt from the user or blank prompt -- redirecting')
         return redirect(url_for('index'))
 
 if __name__ == '__main__':

app.py

@@ -1,10 +1,11 @@
 import json
+import pandas as pd
 from dataclasses import dataclass
 
 from azure.kusto.data import KustoClient
 from azure.kusto.data.exceptions import KustoServiceError
 from azure.kusto.data.helpers import dataframe_from_result_table
-from utils import AuthenticationModeOptions, Utils
+from apis import AuthenticationModeOptions, Utils
 
 @dataclass
 class ConfigJson:
@@ -47,28 +48,15 @@ def load_configs(cls, config_file_name: str) -> None:
             Utils.error_handler(f"Couldn't read config file '{config_file_name}'", ex)
 
     @classmethod
-    def query_table(cls, kusto_client: KustoClient, database_name: str, table_name: str) -> None:
+    def query_table(cls, kusto_client: KustoClient, database_name: str, table_name: str, command: str) -> pd.DataFrame:
         """
         Execute queries on the Kusto table.
         :param kusto_client: Client to run queries
         :param database_name: DB Name
         :param table_name: Table Name
         """
+        df = pd.DataFrame()
         try:
-            escaped_table = f"['{table_name}']"
-            
-            # Get row count
-            # cls.wait_for_user_to_proceed(f"Get row count for '{database_name}.{table_name}':")
-            command = f"{escaped_table} | count"
-            print(f"Executing command: {command}")
-            response = kusto_client.execute_query(database_name, command)
-            if response and response.primary_results:
-                count = response.primary_results[0][0][0]
-                print(f"Row count: {count}")
-
-            # Get sample rows
-            # cls.wait_for_user_to_proceed(f"Get first two rows from '{database_name}.{table_name}':")
-            command = f"{escaped_table} | take 2"
             print(f"Executing command: {command}")
             response = kusto_client.execute_query(database_name, command)
             if response and response.primary_results:
@@ -84,6 +72,7 @@ def query_table(cls, kusto_client: KustoClient, database_name: str, table_name:
             print(f"Error: {ex}")
             print(f"Failed command: {command}")
             raise ex
+        return df
 
     @classmethod
     def wait_for_user_to_proceed(cls, prompt_msg: str) -> None:

kustoQuery.py

@@ -1,4 +1,6 @@
 Flask==3.1.0
 gunicorn
 azure-kusto-data>=0.0.15
-pandas>=1.0.0
+pandas>=1.0.0
+utils
+openai

requirements.txt

@@ -51,11 +51,11 @@ <h1 class="display-6 fw-bold text-primary">XInsights</h1>
             <div class="row">
                 <!-- Query Input Section -->
                 <div class="col-md-8 mx-auto mb-4">
-                    <form method="post" action="{{url_for('hello')}}">
+                    <form method="post" action="{{url_for('execute_prompt')}}">
                         <div class="card">
                             <div class="card-body">
                                 <h5 class="card-title mb-3">Enter Your Query</h5>
-                                <textarea class="form-control mb-3" id="name" name="name" rows="3" placeholder="Type or select a query..."></textarea>
+                                <textarea class="form-control mb-3" id="prompt" name="prompt" rows="3" placeholder="Type or select a query..."></textarea>
                                 <button type="submit" class="btn btn-primary">Submit Query</button>
                             </div>
                         </div>

templates/index.html

@@ -48,10 +48,10 @@
                 <div class="col-md-10 mx-auto">
                     <div class="card mb-4">
                         <div class="card-header bg-primary text-white">
-                            <h5 class="card-title mb-0">Your Query</h5>
+                            <h5 class="card-title mb-0">Response</h5>
                         </div>
                         <div class="card-body">
-                            <p class="card-text">{{name}}</p>
+                            {{output}}
                         </div>
                     </div>