Update app.py

app.py

@@ -126,7 +126,7 @@ def get_schools():
 
     return schools
 
-def get_sports  ():
+def get_sports():
 
     airtable_endpoint = f'https://api.airtable.com/v0/{base_id}/{sports_table_name}'
 
@@ -156,7 +156,7 @@ def get_sports  ():
             # Print the list of 'school' values
             # print(sports)
             else:
-                print("No records found in the 'policies' table.")
+                print("No records found in the 'schools' table.")
         else:
             print(f"Failed to retrieve school data. Status code: {response.status_code}")
     except Exception as e:
@@ -164,6 +164,44 @@ def get_sports  ():
 
     return sports
 
+def get_interests():
+
+    airtable_endpoint = f'https://api.airtable.com/v0/{base_id}/{interests_table_name}'
+
+    # Parameters for the API request to select only the 'interest' field
+    params = {
+        'fields[]': 'interest',  # Replace with the name of your field
+        'sort[0][field]': 'interest',  # Sort by the 'school' field
+        'sort[0][direction]': 'asc',  # Sort in ascending order
+    }
+
+    interests = ''
+
+    try:
+        # Send a GET request to the Airtable API
+        response = requests.get(airtable_endpoint, headers=headers, params=params)
+
+        # Check if the request was successful (status code 200)
+        if response.status_code == 200:
+            # Parse the JSON response
+            data = response.json()
+
+            # Check if there are records in the response
+            if data.get('records'):
+                # Extract the 'school' values from each record
+                interests = [record['fields']['interest'] for record in data['records']]
+
+            else:
+                print("No records found in the 'interests' table.")
+        else:
+            print(f"Failed to retrieve interest data. Status code: {response.status_code}")
+    except Exception as e:
+        print(f"An error occurred: {str(e)}")
+
+    return interests
+
+
+
 def get_prompt(header, template_content):
 
     airtable_endpoint = f'https://api.airtable.com/v0/{base_id}/{prompts_table_name}'    
@@ -275,6 +313,9 @@ def chatbot(school_selection,gender,social_media_channel,sports,interests,output
 
     trimmed_input = prompt_trim(analysis_input)
 
+    with open('analysis_input.txt', 'w', encoding='utf-8') as out_file:
+        out_file.write(trimmed_input)
+  
     response = openai.ChatCompletion.create(
         model="gpt-4",
         #model="gpt-3.5-turbo",
@@ -299,17 +340,20 @@ def chatbot(school_selection,gender,social_media_channel,sports,interests,output
     prompt_tokens = tokens_used.get('prompt_tokens',)
     completion_tokens = tokens_used.get('completion_tokens', 0)
 
+    with open('response.txt', 'w', encoding='utf-8') as out_file:
+        out_file.write(gpt_response)
+
     end_time = datetime.datetime.now()
 
     response_time = end_time - start_time
 
-    global scout_response
+    #global scout_response
 
     scout_response = gpt_response
 
     append_to_at_scountlog(school_selection, output_format, gender,interests,sports,gpt_response,number_of_athletes,social_media_channel,response_time,question_cost,prompt_tokens,completion_tokens)
 
-    return response_label,gpt_response
+    return scout_response
 
 def log_login(username):
 
@@ -326,7 +370,7 @@ def log_login(username):
         }
     try:
 
-        # Post data to Airtable
+    # Post data to Airtable
         response = requests.post(airtable_endpoint, headers=headers, json=data)
 
         # Check for errors
@@ -374,9 +418,10 @@ school_selection = []
 
 schools = get_schools()
 sports = get_sports()
+interests = get_interests()
 
 get_examples()
-
+scout_response = ''
 logged_in_user = 'admin'
 
 with gr.Blocks(CIMStheme) as iface:
@@ -389,26 +434,41 @@ with gr.Blocks(CIMStheme) as iface:
             gr.Markdown("")
     with gr.Row():
         with gr.Column():
-            gr.Interface(fn=chatbot,
-                     inputs=[
-                         gr.components.Dropdown(schools,multiselect=True,info="Select one or more schools. This will select avaliable Student Athletes from the school(s)",label="School Selection"),
-                         gr.components.Dropdown(["Male","Female"],multiselect=True,info="Select one or more.",label="Gender"),
-                         gr.components.Dropdown(["Facebook","Instagram","TikTok","X (Twitter)","YouTube","SnapChat","Twitch"],multiselect=True,info="Select one or more.",label="Social Media Channel"),
-                         gr.components.Dropdown(sports,multiselect=True,info="Select one or more.",label="Sports"),
-                         gr.components.Dropdown(["Fashion","Sports","Food","Cooking","Travel","Arts","Music"],multiselect=True,info="Select one or more.",label="Interests"),
-                         gr.components.Dropdown(["Summary","Detailed Analysis","Table"],multiselect=False,info="Select the desired output format.",label="Output Format"),
-                        gr.components.Dropdown(["1","2","3","4","5","6","7","8","9","10"],multiselect=False,info="How many athletes are you looking for?",label="Number of Athletes")],
-                     outputs=[
-                         gr.components.Markdown(response_label),
-                         gr.components.HTML(label="Scout Response")
-                         ],
-                     description="Use Scout to find and rank Athletes that meet your criteria",
-                         allow_flagging="never",
-                     cache_examples=False
-                         )
+             school_selection=gr.components.Dropdown(schools,multiselect=True,info="Select one or more schools. This will select avaliable Student Athletes from the school(s)",label="School Selection")
+        with gr.Column():
+             gender=gr.components.Dropdown(["Male","Female"],multiselect=True,info="Select one or more.",label="Gender")
+        with gr.Column():
+             social_media_channel=gr.components.Dropdown(["Facebook","Instagram","TikTok","X (Twitter)","YouTube","SnapChat","Twitch"],multiselect=True,info="Select one or more.",label="Social Media Channel")
+    with gr.Row():
+        with gr.Column():
+             sports=gr.components.Dropdown(sports,multiselect=True,info="Select one or more.",label="Sports")
+        with gr.Column():
+             interests=gr.components.Dropdown(interests,multiselect=True,info="Select one or more.",label="Interests")
+        with gr.Column():
+             number_of_athletes=gr.components.Dropdown(["1","2","3","4","5","6","7","8","9","10"],multiselect=False,info="How many athletes are you looking for?",label="Number of Athletes")
+    with gr.Row():
+        with gr.Column():
+         output_format=gr.components.Dropdown(["Summary","Detailed Analysis","Table"],multiselect=False,info="Select the desired output format.",label="Output Format")
+        with gr.Column():
+            gr.Markdown("")
+        with gr.Column():
+            gr.Markdown("")
+    with gr.Row():
+        with gr.Column():
+            btn = gr.components.Button(value="Submit",size='sm')
+        with gr.Column():
+            gr.Markdown("")
+        with gr.Column():
+            gr.Markdown("")
+    with gr.Row():
+        with gr.Column():
+            gr.components.Markdown(response_label)
+    with gr.Row():
+        with gr.Column():
+            scout_response = gr.components.HTML(value="",label="response")
+            btn.click(chatbot, inputs=[school_selection,gender,social_media_channel,sports,interests,output_format,number_of_athletes], outputs=[scout_response])
     with gr.Row():
         with gr.Column():
             gr.HTML('<center><i>CIMS.AI Confidential 2023</i></center>')
 
-iface.launch(auth=login_auth, auth_message= "Enter your username and password that you received from CIMS.AI. To request a login, please email 'info@cims.ai'",share=True)
-#iface.launch()
+iface.launch(auth=login_auth, auth_message= "Enter your username and password that you received from CIMS.AI. To request a login, please email 'info@cims.ai'")