Update app.py

app.py

@@ -72,20 +72,9 @@ def generate(
 
 image_path = "AvvoVhat.png"
 
-with gr.Blocks() as demo:
-    gr.Markdown("# AvvoChat")
-    gr.Markdown("Fai una domanda riguardante la legge italiana all'AvvoChat e ricevi una spiegazione semplice al tuo dubbio.")
-
-    with gr.Row():
-        msg = gr.Textbox(show_label=False, placeholder="Inserisci il tuo messaggio...")
-        btn = gr.Button("Invia")
-
-    with gr.Row():
-        with gr.Column(scale=0.5, min_width=100):
-            gr.Image(image_path, width=50, height=200)
-        with gr.Column(scale=6):
-            chatbox = gr.Chatbot(label="AvvoChat")
 
+chat_interface = gr.ChatInterface(
+    fn = generate
     additional_inputs = [
         gr.Slider(
             label="Max new tokens",
@@ -134,48 +123,33 @@ with gr.Blocks() as demo:
         ["Una persona single può adottare un bambino?"],
         ["Posso usare un immagine creada con l'intelligenza artificiale?"],
         ["Se il mio pallone da calcio cade in un giardino di un'abitazione privata, poss scavalcare il concello per riprendermelo?"],
-    ]
+    ]  
+)
 
-    def on_send_message(message, chat_history, max_new_tokens, temperature, top_p, top_k, do_sample, repetition_penalty):
-        response_stream = generate(
-            message=message,
-            chat_history=chat_history,
-            max_new_tokens=max_new_tokens,
-            temperature=temperature,
-            top_p=top_p,
-            top_k=top_k,
-            do_sample=do_sample,
-            repetition_penalty=repetition_penalty,
-        )
-        #full_response = response_stream[-1]
-        full_response = ""
-        for response in response_stream:
-            full_response += response
-        chat_history.append((message, full_response))
-        return chat_history, ""
-
-    # Collegare il pulsante e la textbox alla funzione on_send_message
-    btn.click(
-        on_send_message,
-        inputs=[
-            msg,
-            chatbox,
-            *additional_inputs
-        ],
-        outputs=[chatbox, msg],
-    )
-    msg.submit(
-        on_send_message,
-        inputs=[
-            msg,
-            chatbox,
-            *additional_inputs
-        ],
-        outputs=[chatbox, msg],
-    )
 
-    for example in examples:
-        gr.Examples(examples=[example], inputs=msg)
+
+
+
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# AvvoChat")
+    gr.Markdown("Fai una domanda riguardante la legge italiana all'AvvoChat e ricevi una spiegazione semplice al tuo dubbio.")
+
+    with gr.Row():
+        msg = gr.Textbox(show_label=False, placeholder="Inserisci il tuo messaggio...")
+        btn = gr.Button("Invia")
+
+    with gr.Row():
+        with gr.Column(scale=0.5, min_width=100):
+            gr.Image(image_path, width=50, height=200)
+        with gr.Column(scale=6):
+            chatbox = gr.Chatbot(label="AvvoChat")
+    chat_interface.render()
+    
+    
+
+    
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(share=True)