Update app.py

app.py

@@ -13,6 +13,26 @@ def sentiment_analysis(text):
   result = pipe(text)
   return result
 
+def openai_chatbot(prompt):
+    # Set up the OpenAI API client
+    openai.api_key = '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'
+
+    # Set up the model and prompt
+    model_engine = "text-davinci-003"
+
+    # Generate a response
+    completion = openai.Completion.create(
+        engine=model_engine,
+        prompt=prompt,
+        max_tokens=1024,
+        n=1,
+        stop=None,
+        temperature=0.5,
+    )
+
+    response = completion.choices[0].text
+
+    return f'🤖 {response}'
 
 with gr.Blocks() as demo:
     gr.Markdown("Choose the Chinese NLP model you want to use.")
@@ -20,5 +40,10 @@ with gr.Blocks() as demo:
         text_button = gr.Button("proceed")          
         text_button.click(fn=sentiment_analysis,inputs=gr.Textbox(placeholder="Enter a positive or negative sentence here..."),
                  outputs=gr.Textbox(label="Sentiment Analysis"))
+    with gr.Tab("General Chatbot"):
+        text_button = gr.Button("proceed")          
+        text_button.click(fn=openai_chatbot,inputs=gr.Textbox(placeholder="Enter any topic you would like to discuss in Chinese"),
+                 outputs=gr.Textbox(label="Chatbot Response"))
+
 
 demo.launch(inline=False)