changes

app.py

@@ -32,15 +32,14 @@ def gpt3_endpoint(api_key, prompt):
 import gradio as gr
 
 #the first module becomes text1, the second module file1
-def greet(length, name, do, tone): 
+def greet(length, name, do, tone, audience, vision): 
     prompt = f"""
     Write a {length} with {tone} tone for a startup with the following info:
     The name of the startup: {name}
     what does the startup do? {do}
+    audience of the pitch: {audience}
+    vision of the startup: {vision}
     """
-    # audience of the pitch: {audience}
-    # vision of the startup: {vision}
-    # """
     return gpt3_endpoint(gpt3_key, prompt).replace('\n', ' ').strip()
 
 with gr.Blocks() as demo:
@@ -53,10 +52,12 @@ with gr.Blocks() as demo:
     input2 = gr.Text(headers=["q1"], label="What is the name of your Startup?", value='Goliath AI Consulting')
     input3 = gr.Text(headers=["q2"], label="What does your startup do?", value='We build AI Recommendation Systems')
     input4 = gr.Radio(["professional", "funny", "innovative", "challenging"], label="Tone", value='innovative')
+    input5 = gr.Radio(["clients", "VCs", "social media"], label="Pitch Audience", value='VCs')
+    input6 = gr.Text(headers=["q5"], label="What is the vision of your company?", value='empowering companies with the latest AI')
     
     btn = gr.Button(value="Write a Pitch!")
     state = gr.Text(label="Output")
-    btn.click(greet, [input1, input2, input3, input4], [state])
+    btn.click(greet, [input1, input2, input3, input4, input5, input6], [state])
 demo.launch(share=False)
 
 # iface = gr.Interface(
@@ -67,8 +68,7 @@ demo.launch(share=False)
 #         gr.Text(headers=["q1"], label="What is the name of your Startup?", value='Goliath AI Consulting'),
 #         gr.Text(headers=["q2"], label="What does your startup do?", value='We build AI Recommendation Systems'),
 #         gr.Radio(["professional", "funny", "innovative", "challenging"], label="Tone", value='innovative'),
-#         # gr.Radio(["clients", "VCs", "social media"], label="Pitch Audience", value='VCs')
-#         # gr.Text(headers=["q5"], label="What is the vision of your company?", value='empowering companies with the latest AI')
+#         
 #     ],
 #     outputs='text'
 # )