Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import logging
+import feedparser
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
@@ -231,9 +232,6 @@ def chat_interface(user_input, history, web_search, decoding_strategy, temperatu
     # Ensure the tokenizer is accessible within the function scope
     global tokenizer
     
-    # Ensure the input is correctly formatted as a dictionary
-    user_input = {"text": user_input}
-    
     # Perform model inference
     response = model_inference(
         user_prompt=user_input,
@@ -246,10 +244,10 @@ def chat_interface(user_input, history, web_search, decoding_strategy, temperatu
         tokenizer=tokenizer  # Pass tokenizer to the model_inference function
     )
     
-    # Update chat history
-    history.append([user_input["text"], response])
+    # Update the chat history with the new interaction
+    history.append([user_input, response])
     
-    # Return the updated history
+    # Return the updated history and the response
     return history, response
 
 # Define the Gradio interface components
@@ -257,7 +255,7 @@ interface = gr.Interface(
     fn=chat_interface,
     inputs=[
         gr.Textbox(label="User Input", placeholder="Type your message here..."),
-        gr.State([], label="Chat History"),
+        gr.State([]),  # Chat history
         gr.Checkbox(label="Perform Web Search"),
         gr.Radio(["Greedy", "Top P Sampling"], label="Decoding strategy"),
         gr.Slider(minimum=0.0, maximum=2.0, step=0.05, label="Sampling temperature", value=0.5),
@@ -265,11 +263,10 @@ interface = gr.Interface(
         gr.Slider(minimum=0.01, maximum=5.0, step=0.01, label="Repetition penalty", value=1),
         gr.Slider(minimum=0.01, maximum=0.99, step=0.01, label="Top P", value=0.9)
     ],
-    outputs=[
-        gr.Chatbot(label="Assistant Response"),
-        gr.State([])
-    ],
-    live=False
+    outputs=[gr.State([]), gr.Textbox(label="Assistant Response")],
+    live=True,
+    title="OpenGPT-4o-Chatty",
+    description="Chat with the AI and optionally perform web searches to enhance responses."
 )
 
 # Launch the Gradio interface