Update app.py

app.py

@@ -188,12 +188,21 @@ def append_to_at_qalog(your_role, school_selection, output_format, input_text, g
         'fields': new_fields
         }
 
+    try:
     # Post data to Airtable
-    response = requests.post(AIRTABLE_ENDPOINT, headers=HEADERS, json=data)
+    response = requests.post(airtable_endpoint, headers=headers, json=data)
 
     # Check for errors
     response.raise_for_status()
 
+    except requests.exceptions.HTTPError as http_error:
+        # Handle the HTTP error (e.g., log it or display an error message)
+        print(f"HTTP error occurred: {http_error}")
+
+    except Exception as e:
+        # Handle exceptions, log errors, or raise them as needed
+        print(f"An error occurred: {str(e)}")
+
 #Chatbot Function
 def chatbot(your_role,school_selection,output_format,input_text):
 
@@ -210,20 +219,15 @@ def chatbot(your_role,school_selection,output_format,input_text):
     header = ''
 
     header, template_content = get_prompt(header, template_content)
-
-    #print(header)
-    #print(template_content)
-
-        
+       
     # Create a Jinja2 template from the content
     template = Template(template_content)
 
     # Render the template with the policy JSON
     analysis_input = template.render(policies=policies, question=input_text,format=output_format)
-
   
-    with open('analysis_input.txt', 'w', encoding='utf-8') as out_file:
-        out_file.write(analysis_input)
+    #with open('analysis_input.txt', 'w', encoding='utf-8') as out_file:
+        #out_file.write(analysis_input)
   
     response = openai.ChatCompletion.create(
         model="gpt-4",
@@ -244,23 +248,13 @@ def chatbot(your_role,school_selection,output_format,input_text):
     gpt_response = response.choices[0].message["content"]
 
     tokens_used = response.usage
-
     question_cost = (tokens_used.get('total_tokens', 0) / 1000) * .03
     prompt_tokens = tokens_used.get('prompt_tokens',)
     completion_tokens = tokens_used.get('completion_tokens', 0)
-
-    #print(question_cost)
-
-    """
-    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
 
-    #append_to_at_qalog(your_role, school_selection, output_format, input_text, gpt_response,response_time,question_cost,prompt_tokens,completion_tokens)
+    append_to_at_qalog(your_role, school_selection, output_format, input_text, gpt_response,response_time,question_cost,prompt_tokens,completion_tokens)
 
     return response_label,gpt_response
 
@@ -277,17 +271,23 @@ def log_login(username):
         'fields': new_fields
         }
 
-    # Post data to Airtable
-    response = requests.post(AIRTABLE_ENDPOINT, headers=HEADERS, json=data)
-
-    # Check for errors
-    response.raise_for_status()
+    try:
+        # Post data to Airtable
+        response = requests.post(AIRTABLE_ENDPOINT, headers=HEADERS, json=data)
     
+        # Check for errors
+        response.raise_for_status()
+        
+   except requests.exceptions.HTTPError as http_error:
+        # Handle the HTTP error (e.g., log it or display an error message)
+        print(f"HTTP error occurred: {http_error}")
 
+    except Exception as e:
+        # Handle exceptions, log errors, or raise them as needed
+        print(f"An error occurred: {str(e)}")
 
 def login_auth(username, password):
 
-
     AIRTABLE_ENDPOINT = f'https://api.airtable.com/v0/{base_id}/{users_table_name}'
     
     # Query the 'users' table to check for a match with the provided username and password
@@ -297,8 +297,6 @@ def login_auth(username, password):
 
     response = requests.get(AIRTABLE_ENDPOINT, headers=HEADERS, params=params)
 
-    print(response)
-    
     if response.status_code == 200:
         data = response.json()
         if data.get('records'):
@@ -308,7 +306,9 @@ def login_auth(username, password):
             logged_in_user = username
             
             return True
-
+            
+        print(f"Invalid user/password combination")
+    
     return False
 
 #Gradio UI