Update app.py

app.py

@@ -27,7 +27,7 @@ def enhance_response(resp, message):
 def chat(user_input, history):
     context = format_context(history) + f"You: {user_input}\n𝕴 𝖆𝖒 𝖍𝖎𝖒:"
     inputs = tokenizer.encode(context, return_tensors="pt", truncation=True, max_length=1024)
-    
+
     outputs = model.generate(
         inputs,
         max_new_tokens=50,
@@ -40,17 +40,12 @@ def chat(user_input, history):
     full_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     response = full_text.split("𝕴 𝖆𝖒 𝖍𝖎𝖒:")[-1].split("\nYou:")[0].strip()
     response = enhance_response(response, user_input)
-    
-    history.append((user_input, response))
-    return history  # Return just the updated chat history
 
-with gr.Blocks() as demo:
-    gr.Markdown("# 𝕴 𝖆𝖒 𝖍𝖎𝖒\n*Smooth • Chill • Emotional*")
-    chatbot = gr.Chatbot(height=400, type="messages", label="Chat")
-    msg = gr.Textbox(placeholder="Type something…", show_label=False)
-    state = gr.State([])
+    history.append((user_input, response))
+    return history
 
-    msg.submit(chat, [msg, state], [chatbot, state])
-    gr.Button("Reset").click(lambda: [], None, [chatbot, state])
+# Create the Gradio Interface
+demo = gr.Interface(fn=chat, inputs=["text", "state"], outputs="state")
 
-demo.launch(server_name="0.0.0.0", server_port=7860, enable_api=True)
+# Launch the app
+demo.launch()