classed

app.py

@@ -3,6 +3,7 @@ import openai
 import gradio as gr
 import requests
 from bs4 import BeautifulSoup
+from conversation import Conversation
 
 try:
     from dotenv import load_dotenv
@@ -12,62 +13,13 @@ except ImportError:
 
 openai.api_key = os.getenv("OPEN_API_KEY")
 
-def get_data(url):
-  if not url.startswith("https://") or url.startswith("http://") or url.startswith("www."):
-     gr.Error("Please enter a valid URL")
-  global messages
-  html = requests.get(url).text
-  doc = BeautifulSoup(html, 'html.parser')
-
-
-  headings_1 = [e.text for e in doc.find_all('h1')]
-  headings_2 = [e.text for e in doc.find_all('h2')]
-  # headings_3 = [e.text for e in doc.find_all('h3')]
-  paragraphs = [e.text for e in doc.find_all('p')]
-  # spans = [e.text for e in doc.find_all('span')]
-  joined_paragraphs = (' '.join(paragraphs))
-
-  if len(joined_paragraphs) > 7500:
-    paragraphs = joined_paragraphs[:5000]
-
-  messages = []
-  messages.append({'role': 'system', 'content': "You are a helpful assistant that must answer questions about a website."})
-  messages.append({'role': 'system', 'content': f"here are the h1s - {headings_1}"})
-  messages.append({'role': 'system', 'content': f"here are the h2s - {headings_2}"})
-  # messages.append({'role': 'system', 'content': f"here are the h3s - {headings_3}"})
-  messages.append({'role': 'system', 'content': f"here are the paragraphs - {paragraphs}"})
-  # messages.append({'role': 'system', 'content': f"here are the spans - {spans}"})
-  return messages
-
-def ask_chatbot(input):
-    if input:
-        messages.append({"role": "user", "content": input})
-        chat = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo", messages=messages
-        )
-        reply = chat.choices[0].message.content
-        messages.append({"role": "assistant", "content": reply})
-        return reply
-    
-
-def user(user_message, history):
-    return "", history + [[user_message, None]]
-
-def bot(history):
-    user_message = history[-1][0]
-    try:
-      bot_message = ask_chatbot(user_message)
-    except NameError:
-      bot_message = "Please try again"
-    history[-1][1] = bot_message
-    return history
-
 with gr.Blocks(css="footer {visibility: hidden}", title="chatWEBPAGE") as demo:
+    conversation = Conversation()
     gr.Markdown("Enter your website url, then ask the AI a question.")
     with gr.Row():
         with gr.Column(scale=2):
             url = gr.Textbox(label="1. Enter a webpage URL to chat about")
-            url.change(fn=get_data, inputs=url)
+            url.change(fn=conversation.get_data, inputs=url)
         with gr.Column(scale=3):
           gr.Examples([
                       "https://www.bbc.com/news/business-64937251",
@@ -76,8 +28,7 @@ with gr.Blocks(css="footer {visibility: hidden}", title="chatWEBPAGE") as demo:
     chatbot = gr.Chatbot().style(height=150)
 
     msg = gr.Textbox(label="2. Chat with AI about the webpage")
-    msg.submit(user, [msg, chatbot], [msg, chatbot]).success(bot, chatbot, chatbot)
-
+    msg.submit(conversation.user, [msg, chatbot], [msg, chatbot]).success(conversation.bot, chatbot, chatbot)
     gr.Examples(["Please summarise the webpage", "What is the tone of the webpage?", "Tell me your favorite part of the webpage"], inputs=[msg])
     
     with gr.Row():
@@ -86,93 +37,9 @@ with gr.Blocks(css="footer {visibility: hidden}", title="chatWEBPAGE") as demo:
             clear_button.click(lambda: None, None, chatbot, queue=False)
         with gr.Column(scale=4):
             submit_button = gr.Button("Submit")
-            submit_button.click(user, [msg, chatbot], [msg, chatbot], queue=False).success(
-                bot, chatbot, chatbot
+            submit_button.click(conversation.user, [msg, chatbot], [msg, chatbot], queue=False).success(
+                conversation.bot, chatbot, chatbot
             )
 
 if __name__ == "__main__":
-    demo.launch()
-
-  # msg.submit(user, [msg, chatbot_item], [msg, chatbot_item], queue=False)
-  # submit_button.click(user, [msg, chatbot_item], [msg, chatbot_item], queue=False)
-  # clear_button.click(bot, chatbot_item, chatbot_item, queue=False)
-
-
-# if __name__ == "__main__":
-
-  # with gr.Row():
-  #     with gr.Column(scale=2):
-  #       inputs = gr.Textbox(lines=7, label="Chat with AI about the webpage")
-  #       gr.Examples(["Please summarise the webpage", "What is the tone of the webpage?", "Tell me your favorite part of the webpage"], inputs=[inputs])
-  #     with gr.Column(scale=4):
-  #       outputs = gr.Textbox(label="Here's the reply from the AI")
-  # with gr.Row():
-  #     greet_btn = gr.Button("Submit")
-  #     greet_btn.click(fn=chatbot, inputs=inputs, outputs=outputs, show_progress=True)
-
-# demo.queue(concurrency_count=3)
-
-  
-# with gr.Blocks() as demo:
-#     chatbot = gr.Chatbot()
-#     msg = gr.Textbox()
-#     clear = gr.Button("Clear")
-
-#     def user(user_message, history):
-#         return "", history + [[user_message, None]]
-
-#     def bot(history):
-#         bot_message = random.choice(["Yes", "No"])
-#         history[-1][1] = bot_message
-#         time.sleep(1)
-#         return history
-
-#     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-#         bot, chatbot, chatbot
-#     )
-#     clear.click(lambda: None, None, chatbot, queue=False)
-
-# if __name__ == "__main__":
-#     demo.launch()
-
-# interface = gr.Interface(fn=lambda x: x, inputs=demo, outputs=None, theme="compact")
-# # interface.launch()
-
-# url = gr.Textbox(label="Enter a webpage URL to chat about")
-# # save_btn = gr.Button("Save URL")
-# # save_btn.click(fn=get_data, inputs=url, outputs=update_button(save_btn))
-# gr.Examples(["https://www.nerdwallet.com/article/travel/how-much-are-my-united-miles-worth",
-#               "https://www.bbc.com/news/business-64937251",
-#               "https://www.ycombinator.com/"], inputs=[url])
-# inputs = gr.inputs.Textbox(lines=7, label="Chat with AI about the webpage")
-# outputs = gr.outputs.Textbox(label="Here's the reply from the AI")
-
-# iface = gr.Interface(
-#     chatbot,
-#     [url, inputs],
-#     [outputs],
-#     live=True,
-# )
-# iface.launch()
-
-
-# url_box = gr.Textbox(label="Url")
-# inputs = gr.inputs.Textbox(lines=7, label="Chat with AI")
-# outputs = gr.outputs.Textbox(label="Reply")
-
-# gr.Interface(fn=chatbot, inputs=[url_box,inputs], outputs=outputs, title="CHATwebpage AI Chatbot",
-#              description="Ask anything you want",
-#              theme="compact").launch()
-
-# with gr.Blocks() as demo:
-#     used_letters_var = gr.State([])
-
-#     url = gr.Textbox(label="Url")
-#     save_btn = gr.Button("Save")
-#     save_btn.click(fn=get_data, inputs=url, outputs)
-#     # inputs = gr.inputs.Textbox(lines=7, label="Chat with AI")
-#     # outputs = gr.outputs.Textbox(label="Reply")
-#     # greet_btn = gr.Button("Submit")
-#     # greet_btn.click(fn=chatbot, inputs=inputs, outputs=outputs)
-
-# demo.launch()
+    demo.launch()

conversation.py

@@ -0,0 +1,76 @@
+import os
+import openai
+import gradio as gr
+import requests
+from bs4 import BeautifulSoup
+
+try:
+    from dotenv import load_dotenv
+    load_dotenv()
+except ImportError:
+    pass  # In production, python-dotenv may not be installed
+
+openai.api_key = os.getenv("OPEN_API_KEY")
+
+class Conversation:
+    def __init__(self):
+        self.messages = []
+
+    def get_data(self, url):
+        # ... your existing get_data implementation ...
+        # Replace `messages` with `self.messages`
+
+        if not url.startswith("https://") or url.startswith("http://") or url.startswith("www."):
+            gr.Error("Please enter a valid URL")
+        self.messages
+        html = requests.get(url).text
+        doc = BeautifulSoup(html, 'html.parser')
+
+
+        headings_1 = [e.text for e in doc.find_all('h1')]
+        headings_2 = [e.text for e in doc.find_all('h2')]
+        # headings_3 = [e.text for e in doc.find_all('h3')]
+        paragraphs = [e.text for e in doc.find_all('p')]
+        # spans = [e.text for e in doc.find_all('span')]
+        joined_paragraphs = (' '.join(paragraphs))
+
+        if len(joined_paragraphs) > 7500:
+          paragraphs = joined_paragraphs[:5000]
+
+        self.messages = []
+        self.messages.append({'role': 'system', 'content': "You are a helpful assistant that must answer questions about a website."})
+        self.messages.append({'role': 'system', 'content': f"here are the h1s - {headings_1}"})
+        self.messages.append({'role': 'system', 'content': f"here are the h2s - {headings_2}"})
+        # messages.append({'role': 'system', 'content': f"here are the h3s - {headings_3}"})
+        self.messages.append({'role': 'system', 'content': f"here are the paragraphs - {paragraphs}"})
+        # messages.append({'role': 'system', 'content': f"here are the spans - {spans}"})
+        return self.messages
+
+    def ask_chatbot(self, input):
+        # ... your existing ask_chatbot implementation ...
+        # Replace `messages` with `self.messages`
+        if input:
+            self.messages.append({"role": "user", "content": input})
+            chat = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo", messages=self.messages
+            )
+            reply = chat.choices[0].message.content
+            self.messages.append({"role": "assistant", "content": reply})
+            return reply
+
+
+    def user(self, user_message, history):
+        # ... your existing user implementation ...
+        # Replace `messages` with `self.messages`
+        return "", history + [[user_message, None]]
+
+    def bot(self, history):
+        # ... your existing bot implementation ...
+        # Replace `messages` with `self.messages`
+        user_message = history[-1][0]
+        try:
+          bot_message = self.ask_chatbot(user_message)
+        except NameError:
+          bot_message = "Please try again"
+        history[-1][1] = bot_message
+        return history