Update app.py

app.py

@@ -11,11 +11,55 @@ import plotly
 import gradio as gr
 
 # Initiate functions to be defined with API key input.
-def prompt_analyze_questions():
-    pass
-
-def prompt_analyze_reporting():
-    pass
+data_structure_overview = ''
+data_relationships_overview = ''
+fig =''
+output_string = ''
+
+
+def prompt_analyze_reporting(prompt):
+    output = openai.ChatCompletion.create(model="gpt-3.5-turbo",temperature = 0.0, messages=[{"role": "user", "content": 
+                                                               data_structure_overview},
+                                                                       {"role": "user", "content": 
+                                                               data_relationships_overview},{"role": "user", "content": 
+                                                               f"""Do not attempt to use .csv files in your code."""},
+                                                                {"role": "user", "content": 
+                                                               f"""Only use plotly to output charts, graphs, or figures. Do not use matplotlib or other charting libraries. Name the chart object as 'fig'"""},
+                                                                       {"role": "user", "content": 
+                                                               f"""Create a python script to: {prompt}"""}
+                                                                      ])
+    global parsed_response
+    parsed_response = output.choices[0].message.content.strip().split('```python')[len(output.choices[0].message.content.strip().split('```python')) -1 ].split('```')[0]
+    parsed_response_global = f"""global fig 
+global string
+    {parsed_response}"""
+    exec(parsed_response_global)
+    return fig
+
+def prompt_analyze_questions(prompt):
+    output = openai.ChatCompletion.create(model="gpt-3.5-turbo",temperature = 0.0, messages=[{"role": "user", "content": 
+                                                               data_structure_overview},
+                                                                       {"role": "user", "content": 
+                                                               data_relationships_overview},{"role": "user", "content": 
+                                                               f"""Do not attempt to use .csv files in your code."""},
+                                                                {"role": "user", "content": 
+                                                               f"""Do not attempt to create charts or visualize the question with graphics. Only provide string responses."""},
+                                                                {"role": "user", "content": 
+                                                               f"""If you are asked to create visualizations or graphs, create a python script to store a string variable named output_string with the text 'Sorry, I cannot create reporting, select 'Add Reporting' to create reports."""},
+                                                                       {"role": "user", "content": 
+                                                               f"""Create a python script to: {prompt}"""},    
+                                                                {"role": "user", "content": 
+                                                               f"""Store the final response as a string variable named output_string"""}
+                                                                      ])
+
+    global parsed_response
+    parsed_response = output.choices[0].message.content.strip().split('```python')[len(output.choices[0].message.content.strip().split('```python')) -1 ].split('```')[0]
+    parsed_response_global = f"""global fig 
+global string
+{parsed_response}
+globals().update(locals())"""
+    exec(parsed_response_global)
+    return output_string
 
 # Data Retrieval
 
@@ -216,50 +260,6 @@ def getData(tlspc_api_key, openai_api_key):
         The column values in the 'certificateIssuingTemplateId' column in certs_request_df match the column values in the 'issuing_template_id' column in issuing_templates_df.
         """
         
-        def prompt_analyze_reporting(prompt):
-            output = openai.ChatCompletion.create(model="gpt-3.5-turbo",temperature = 0.0, messages=[{"role": "user", "content": 
-                                                                       data_structure_overview},
-                                                                               {"role": "user", "content": 
-                                                                       data_relationships_overview},{"role": "user", "content": 
-                                                                       f"""Do not attempt to use .csv files in your code."""},
-                                                                        {"role": "user", "content": 
-                                                                       f"""Only use plotly to output charts, graphs, or figures. Do not use matplotlib or other charting libraries. Name the chart object as 'fig'"""},
-                                                                               {"role": "user", "content": 
-                                                                       f"""Create a python script to: {prompt}"""}
-                                                                              ])
-            global parsed_response
-            parsed_response = output.choices[0].message.content.strip().split('```python')[len(output.choices[0].message.content.strip().split('```python')) -1 ].split('```')[0]
-            parsed_response_global = f"""global fig 
-        global string
-            {parsed_response}"""
-            exec(parsed_response_global)
-            return fig
-        
-        def prompt_analyze_questions(prompt):
-            output = openai.ChatCompletion.create(model="gpt-3.5-turbo",temperature = 0.0, messages=[{"role": "user", "content": 
-                                                                       data_structure_overview},
-                                                                               {"role": "user", "content": 
-                                                                       data_relationships_overview},{"role": "user", "content": 
-                                                                       f"""Do not attempt to use .csv files in your code."""},
-                                                                        {"role": "user", "content": 
-                                                                       f"""Do not attempt to create charts or visualize the question with graphics. Only provide string responses."""},
-                                                                        {"role": "user", "content": 
-                                                                       f"""If you are asked to create visualizations or graphs, create a python script to store a string variable named output_string with the text 'Sorry, I cannot create reporting, select 'Add Reporting' to create reports."""},
-                                                                               {"role": "user", "content": 
-                                                                       f"""Create a python script to: {prompt}"""},    
-                                                                        {"role": "user", "content": 
-                                                                       f"""Store the final response as a string variable named output_string"""}
-                                                                              ])
-
-            global parsed_response
-            parsed_response = output.choices[0].message.content.strip().split('```python')[len(output.choices[0].message.content.strip().split('```python')) -1 ].split('```')[0]
-            parsed_response_global = f"""global fig 
-        global string
-        {parsed_response}
-        globals().update(locals())"""
-            exec(parsed_response_global)
-            return output_string
-        
         # Store variables for use in other portions of the application
         globals().update(locals())