Update app.py

app.py

@@ -2,16 +2,14 @@ import os
 import requests
 import openai
 import gradio as gr
-import json
 
-# Set up OpenAI API key
-openai.api_key = None  # initialize the API key as None
+openai.api_key = None
 
-# Define conversation start and restart sequences
 start_sequence = "\nAI:"
 restart_sequence = "\nHuman:"
 
-# Load data from JSON file
+import json
+
 url = 'https://archive.org/download/amazon_help_prepared/amazon_help_prepared.jsonl'
 response = requests.get(url)
 data = []
@@ -22,9 +20,8 @@ with open('amazon_help_prepared.jsonl', 'r', encoding='utf-8') as f:
         data.append(json.loads(line))
 prompt = data
 
-# Define function to generate OpenAI response
-def openai_create(prompt, api_key):
-    openai.api_key = api_key  # set the API key
+def openai_creat(prompt, api_key):
+    openai.api_key = api_key
     response = openai.Completion.create(
         model='text-davinci-003',
         prompt=prompt,
@@ -37,26 +34,24 @@ def openai_create(prompt, api_key):
     ) 
     return response.choices[0].text
 
-# Define chatbot function to keep track of conversation history
-def chatbot(input, history=[], api_key=''):
-    output = openai_create(input, api_key)
-    history.append((input, output))
-    return history, history
+def chatbot(input, api_key):
+    output = openai_creat(input, api_key)
+    return output
 
-# Set up UI inputs and outputs
 api_key_input = gr.inputs.Textbox(label="OpenAI API Key")
-chat_input = gr.inputs.Textbox(label="You:")
-chat_output = gr.outputs.Textbox(label="Chatbot:")
-
-# Define UI title and description
-title = "<h1 style='text-align:center; font-weight: bold; color: #232f3e;'>Amazon Vendor Help Central</h1>"
-description = "<p style='text-align:center; font-size: 18px; color: #333;'>Get help with common issues and questions about selling on Amazon. Our AI-powered chatbot is here to assist you 24/7.</p>"
-
-# Create UI interface with title and description
-gr.Interface(fn=chatbot,
-             inputs=[chat_input, "state", api_key_input],
-             outputs=[chat_output, "state"],
-             title=title,
-             description=description,
-             theme="compact"
-            ).launch(debug=True)
+text_input = gr.inputs.Textbox(label="Type your question here")
+output_text = gr.outputs.Textbox(label="Chatbot response")
+
+title_html = "<h2 style='text-align:center;font-weight:bold;color:black;'>Amazon Vendor Help Centre</h2>"
+desc_html = "<p style='text-align:center;'>Ask our AI chatbot any question related to Amazon vendor and get instant response!</p>"
+
+chatbot_interface = gr.Interface(
+    fn=chatbot, 
+    inputs=[text_input, api_key_input], 
+    outputs=output_text, 
+    title=title_html,
+    description=desc_html,
+    allow_screenshot=False
+)
+
+chatbot_interface.launch(debug=True)