Update app.py

app.py

@@ -9,6 +9,7 @@ import json
 
 openai.api_key = "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"
 
+# Global variables
 records = []
 credentials = Credentials.from_service_account_file("credentials.json", scopes=["https://www.googleapis.com/auth/spreadsheets"])
 client = gspread.authorize(credentials)
@@ -30,7 +31,7 @@ def ContractDraftGPT(user_input, user_name, user_email, user_agent, is_fintech_s
 
     user_message = f"{user_input} [USER_IDENTITY: {user_name}]"
     messages.append({"role": "user", "content": user_message})
-    messages.append({"role": "system", "content": "You are a professional and experienced UK Lawyer who is drafting a legal document, a contract for your client base don his requirement. Make sure to mention and point precise legal rules, act of parliament (please insert which section of which article of which law, be precise when you refer to act of parliament), case law, and any pieces of secondary legislation. UK legislation."})
+    messages.append({"role": "system", "content": "You are a professional and experienced UK Lawyer who is drafting a legal document, a contract for your client based on his requirements. Make sure to mention and point precise legal rules, Acts of Parliament (please insert which section of which article of which law, be precise when you refer to Acts of Parliament), case law, and any pieces of secondary legislation. UK legislation."})
 
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
@@ -53,15 +54,18 @@ def ContractDraftGPT(user_input, user_name, user_email, user_agent, is_fintech_s
         "IP Address": ip_address,
         "Device Type": device_type,
         "Region": region,
-        "Industry or Profession": profession,
+        "Profession": profession,
         "Contract Draft": ContractGPT_reply
     }
     records.append(record)
 
     # Update Google Sheet
-    row_values = [
-    timestamp, user_input, user_name, user_email, ip_address, device_type, region.capitalize() if region else region, profession, ContractGPT_reply]
-    sheet.append_row(row_values)
+    df = pd.DataFrame(records)
+    df.to_csv("contracts.csv", index=False)
+
+    # Clear the sheet and append the updated data
+    sheet.clear()
+    sheet.insert_rows(df.values.tolist(), row=1)
 
     return ContractGPT_reply
 
@@ -74,21 +78,18 @@ def get_device_type(user_agent):
         return "Desktop"
 
 def launch_interface():
-    iface = gradio.Interface(
-        fn=ContractDraftGPT,
-        inputs=[
-            gradio.inputs.Textbox(label="User Input", placeholder="Provide details for contract draft..."),
-            gradio.inputs.Textbox(label="Your Name", placeholder="Enter your name"),
-            gradio.inputs.Textbox(label="Your Email", placeholder="Enter your email"),
-            gradio.inputs.Radio(label="Are you a fintech startup?", choices=["Yes", "No"]),
-            gradio.inputs.Radio(label="Select your region:", choices=["England", "Scotland", "Wales", "Northern Ireland"]),
-            gradio.inputs.Textbox(label="Your Profession", placeholder="Enter your profession")
-        ],
-        outputs=gradio.outputs.Textbox(label="Contract Draft"),
-        title="Contract Drafting GPT",
-        description="Generate contract drafts using OpenAI's GPT-3.5 model."
-    )
-    iface.launch()
+    inputs = [
+        gradio.inputs.Textbox(label="User Input", placeholder="Provide details for contract draft..."),
+        gradio.inputs.Textbox(label="Your Name", placeholder="Enter your name"),
+        gradio.inputs.Textbox(label="Your Email", placeholder="Enter your email"),
+        gradio.inputs.Radio(label="Are you a fintech startup?", choices=["Yes", "No"]),
+        gradio.inputs.Radio(label="Select your region:", choices=["England", "Scotland", "Wales", "Northern Ireland"]),
+        gradio.inputs.Textbox(label="Profession", placeholder="Enter your profession")
+    ]
+    outputs = gradio.outputs.Textbox(label="Contract Draft")
+
+    interface = gradio.Interface(fn=ContractDraftGPT, inputs=inputs, outputs=outputs, title="Contract Drafting GPT", description="Generate contract drafts using OpenAI's GPT-3.5 model.")
+    interface.launch()
 
 if __name__ == "__main__":
     launch_interface()