Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import torch
 import theme
 
-
+DEVEL = os.environ.get('DEVEL', False)
 theme = theme.Theme()
 
 import os
@@ -36,20 +36,7 @@ from langchain_community.document_loaders import WebBaseLoader
 from pydantic import BaseModel
 import shutil
 
-title = """
-<picture>
-  <source srcset="https://huggingface.co/spaces/rocioadlc/test_4/resolve/main/front.jpg" media="(prefers-color-scheme: dark)" />
-  <img alt="WhisperSpeech banner with Collabora and LAION logos" src="https://huggingface.co/spaces/rocioadlc/test_4/resolve/main/front.jpg" media="(prefers-color-scheme: dark)" style="width: 60%; margin: 0 auto;" />
-</picture>
-¡Bienvenido a nuestro clasificador de imágenes y chatbot para un reciclaje más inteligente!
-
-¿Alguna vez te has preguntado si puedes reciclar un objeto en particular? ¿O te has sentido abrumado por la cantidad de residuos que generas y no sabes cómo manejarlos de manera más sostenible? ¡Estás en el lugar correcto!
-Nuestra plataforma combina la potencia de la inteligencia artificial con la comodidad de un chatbot para brindarte respuestas rápidas y precisas sobre qué objetos son reciclables y cómo hacerlo de la manera más eficiente.
-
-¿Cómo funciona?
-Clasificador de Imágenes: Simplemente carga una foto del objeto que quieras reciclar, y nuestro modelo de inteligencia artificial identificará rápidamente el material y te dirá cómo desecharlo adecuadamente.
-Chatbot de Respuestas: ¿Tienes preguntas sobre reciclaje, materiales específicos o prácticas sostenibles? ¡Pregunta a nuestro chatbot! Está aquí para responder todas tus preguntas y ayudarte a tomar decisiones más informadas sobre tu reciclaje.
-"""
+title= "Green Greta"
 
 from huggingface_hub import from_pretrained_keras
 
@@ -91,15 +78,7 @@ def predict_image(image_input):
     
     return category_scores
 
-
-image_gradio_app = gr.Interface(
-    fn=predict_image,
-    inputs=gr.Image(label="Image", sources=['upload', 'webcam'], type="pil"),
-    outputs=[gr.Label(label="Result")],
-    title="Green Greta",
-    theme=theme
-)
-
+#Cell 2: Chatbot
 loader = WebBaseLoader(["https://www.epa.gov/recycle/frequent-questions-recycling", "https://www.whitehorsedc.gov.uk/vale-of-white-horse-district-council/recycling-rubbish-and-waste/lets-get-real-about-recycling/", "https://www.teimas.com/blog/13-preguntas-y-respuestas-sobre-la-ley-de-residuos-07-2022", "https://www.molok.com/es/blog/gestion-de-residuos-solidos-urbanos-rsu-10-dudas-comunes"])
 data=loader.load()
 # split documents
@@ -192,25 +171,41 @@ def chat_interface(question,history):
     return answer_value
 
 
+
+image_gradio_app = gr.Interface(
+    fn=predict_image,
+    inputs=gr.Image(label="Image", sources=['upload', 'webcam'], type="pil"),
+    outputs=[gr.Label(label="Result")],
+    title="Green Greta",
+    theme=theme
+)
+
 chatbot_gradio_app = gr.ChatInterface(
     fn=chat_interface,
     title="Green Greta"
 )
 
-# Combine both interfaces into a single app
+banner_tab_content = """
+<picture>
+  <source srcset="https://huggingface.co/spaces/rocioadlc/test_4/resolve/main/front.jpg" media="(prefers-color-scheme: dark)" />
+  <img alt="WhisperSpeech banner with Collabora and LAION logos" src="https://huggingface.co/spaces/rocioadlc/test_4/resolve/main/front.jpg" media="(prefers-color-scheme: dark)" style="width: 60%; margin: 0 auto;" />
+</picture>
+¡Bienvenido a nuestro clasificador de imágenes y chatbot para un reciclaje más inteligente!
+¿Alguna vez te has preguntado si puedes reciclar un objeto en particular? ¿O te has sentido abrumado por la cantidad de residuos que generas y no sabes cómo manejarlos de manera más sostenible? ¡Estás en el lugar correcto!
+Nuestra plataforma combina la potencia de la inteligencia artificial con la comodidad de un chatbot para brindarte respuestas rápidas y precisas sobre qué objetos son reciclables y cómo hacerlo de la manera más eficiente.
+¿Cómo funciona?
+Clasificador de Imágenes: Simplemente carga una foto del objeto que quieras reciclar, y nuestro modelo de inteligencia artificial identificará rápidamente el material y te dirá cómo desecharlo adecuadamente.
+Chatbot de Respuestas: ¿Tienes preguntas sobre reciclaje, materiales específicos o prácticas sostenibles? ¡Pregunta a nuestro chatbot! Está aquí para responder todas tus preguntas y ayudarte a tomar decisiones más informadas sobre tu reciclaje.
+"""
+
+banner_tab = gr.Markdown(banner_tab_content)
+
+# Combinar ambas interfaces en una sola aplicación con pestañas
 app = gr.TabbedInterface(
-    [image_gradio_app, chatbot_gradio_app],
-    tab_names=["Green Greta Image Classification","Green Greta Chat"],
+    [banner_tab, image_gradio_app, chatbot_gradio_app],
+    tab_names=["Banner & Explanation", "Green Greta Image Classification", "Green Greta Chat"],
     theme=theme
 )
 
-with gr.Blocks() as demo:
-    # Muestra la imagen y el texto
-    gr.Markdown(title)
-    # Agrega espacio vertical entre la imagen y el texto y la siguiente sección
-    gr.Markdown("  \n  ")
-    # Agrega el clasificador de imágenes y el chatbot debajo
-    app.launch()
-    
-demo.queue()
-demo.launch()
+app.queue()
+app.launch()