Update app.py

app.py

@@ -44,7 +44,7 @@ function_descriptions = [
     }
 ]
 
-email = """
+test_email = """
 Dear Jason 
 I hope this message finds you well. I'm Shirley from Gucci;
 
@@ -57,16 +57,26 @@ Looking forward
 Shirley Lou
 """
 
-prompt = f"Please extract key information from this email: {email} "
-message = [{"role": "user", "content": prompt}]
+email = st.text_input("email",test_email)
+
+def extract_data(email,function_descriptions):
+    prompt = f"Please extract key information from this email: {email} "
+    message = [{"role": "user", "content": prompt}]
+    
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=message,
+        functions = function_descriptions,
+        function_call="auto"
+    )
+
+    print(response)
+    return response
+
+if st.button(Extract):
+    data = extract_data(email,function_descriptions)
+    st.json(data)
+
 
-response = openai.ChatCompletion.create(
-    model="gpt-3.5-turbo",
-    messages=message,
-    functions = function_descriptions,
-    function_call="auto"
-)
 
-print(response)
 
-st.header("Done!")