cleanup and openai

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 import random
 import openai
 
-def tell_us_about_yo_momma(category, text):
+def tell_us_about_yo_momma(category, topic):
     openAI_key = os.getenv('my_openAI_key')
     if openAI_key.strip()=='':
         return '[ERROR]: Please enter you Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'
-   
 
+    prompt = ""
+    prompt = f"Tell us a yo momma joke about her {category} and {topic}"
+   
    completions = openai.Completion.create(
         engine=engine,
         prompt=prompt,
@@ -55,10 +57,6 @@ def launch_demo():
         "Movies/TV Shows"
     ]
 
-    def analyze_input(category, text):
-        result = tell_us_about_yo_momma(category, text)
-        return result
-
     dropdown = gr.components.Dropdown(categories, label="Category")
     text_input = gr.components.Textbox(label="Topic")
     output = gr.components.Textbox(label="Lolz")
@@ -78,7 +76,7 @@ def launch_demo():
     ]
 
     gr.Interface(
-        fn=analyze_input,
+        fn=tell_us_about_yo_momma,
         inputs=[dropdown, text_input],
         outputs=output,
         examples=examples,