Spaces:

AlbertoFH98
/

PodCastena

Runtime error

App Files Files Community

AlbertoFH98 commited on Nov 27, 2023

Commit

90e8e7a

1 Parent(s): 2bae7ed

Update app.py

Browse files

Files changed (1) hide show

app.py +73 -4

app.py CHANGED Viewed

@@ -46,11 +46,72 @@ def get_basics_comp(emb_model, model, default_system_prompt_link, _logger, podca
     r    = requests.get("https://raw.githubusercontent.com/AlbertoUAH/Castena/main/media/castena-animated-icon.gif", stream=True)
     icon = Image.open(r.raw)
     icon = icon.resize((img_size, img_size))
-    st.sidebar.image(icon)
     video_option = st.sidebar.selectbox(
         "Seleccione el podcast",
-        list(podcast_url_video_df['podcast_name_lit'].apply(lambda x: x.replace("'", "")))
     )
     video_option_joined = '_'.join(video_option.replace(': Entrevista a ', ' ').lower().split(' ')).replace("\'", "")
     video_option_joined_path = "{}_transcription.txt".format(video_option_joined)
     youtube_video_url   = list(podcast_url_video_df[podcast_url_video_df['podcast_name'].str.contains(video_option_joined)]['youtube_video_url'])[0].replace("\'", "")
@@ -67,7 +128,12 @@ def get_basics_comp(emb_model, model, default_system_prompt_link, _logger, podca
     together.api_key = os.environ["TOGETHER_API_KEY"]
     together.Models.start(model)
     return together, translator, nlp, retriever, video_option, video_option_joined_path, default_system_prompt, youtube_video_url
 def main():
     args, logger = get_args()
     B_INST, E_INST = "[INST]", "[/INST]"
@@ -82,13 +148,13 @@ def main():
     WIDTH                      = 50
     SIDE                       = (100 - WIDTH) / 2
     podcast_url_video_df = get_podcast_data(PODCAST_URL_VIDEO_PATH)
     together, translator, nlp, retriever, video_option, video_option_joined_path, default_system_prompt, youtube_video_url = get_basics_comp(EMB_MODEL, MODEL,
                                                                                                                                              DEFAULT_SYSTEM_PROMPT_LINK, logger,
                                                                                                                                              podcast_url_video_df, img_size=100)
     # -- 6. Setup prompt template + llm chain
     instruction = """CONTEXTO:/n/n {context}/n
@@ -111,9 +177,12 @@ RESPUESTA: """
     if "messages" not in st.session_state:
         st.session_state.messages = []
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
     if prompt := st.chat_input("¡Pregunta lo que quieras!"):
         with st.chat_message("user"):
             st.markdown(prompt)

     r    = requests.get("https://raw.githubusercontent.com/AlbertoUAH/Castena/main/media/castena-animated-icon.gif", stream=True)
     icon = Image.open(r.raw)
     icon = icon.resize((img_size, img_size))
+    #st.sidebar.image(icon)
+    with st.sidebar.container():
+        st.markdown(
+            """
+            <head>
+                <style>
+                    .footer1 {
+                        text-align: center;
+                    }
+                </style>
+            </head>
+            <body>
+                <div class="footer1">
+                    <img src=https://raw.githubusercontent.com/AlbertoUAH/Castena/main/media/castena-animated-icon.gif width="150" height="150">
+                </div>
+                <br>
+            </body>
+            """,
+            unsafe_allow_html=True,
+    )
+    podcast_list = list(podcast_url_video_df['podcast_name_lit'].apply(lambda x: x.replace("'", "")))
     video_option = st.sidebar.selectbox(
         "Seleccione el podcast",
+        podcast_list,
+        on_change=clean_chat
     )
+    # -- Add icons
+    with st.sidebar.container():
+        st.markdown(
+            """
+            <head>
+                <style>
+                    .footer2 {
+                        position: fixed;
+                        bottom: 2%;
+                        left: 6.5%;
+                    }
+                    .footer2 a {
+                        margin: 10px;
+                        text-decoration: none;
+                    }
+                </style>
+            </head>
+            <body>
+                <div class="footer2">
+                    <a href="https://www.linkedin.com/in/alberto-fernandez-hernandez-3a3474136">
+                        <img src="https://cdn-icons-png.flaticon.com/128/3536/3536505.png" width="32" height="32">
+                    </a>
+                    <a href="https://github.com/AlbertoUAH/Castena">
+                        <img src="https://cdn-icons-png.flaticon.com/128/733/733553.png" width="32" height="32">
+                    </a>
+                    <a href="https://www.buymeacoffee.com/castena">
+                        <img src="https://cdn-icons-png.flaticon.com/128/761/761767.png" width="32" height="32">
+                    </a>
+                </div>
+            </body>
+            """,
+            unsafe_allow_html=True,
+    )
     video_option_joined = '_'.join(video_option.replace(': Entrevista a ', ' ').lower().split(' ')).replace("\'", "")
     video_option_joined_path = "{}_transcription.txt".format(video_option_joined)
     youtube_video_url   = list(podcast_url_video_df[podcast_url_video_df['podcast_name'].str.contains(video_option_joined)]['youtube_video_url'])[0].replace("\'", "")
     together.api_key = os.environ["TOGETHER_API_KEY"]
     together.Models.start(model)
     return together, translator, nlp, retriever, video_option, video_option_joined_path, default_system_prompt, youtube_video_url
+def clean_chat():
+    st.session_state.conversation = None
+    st.session_state.chat_history = None
+    st.session_state.messages     = [{'role': 'assistant', 'content': 'Nuevo chat creado'}]
 def main():
     args, logger = get_args()
     B_INST, E_INST = "[INST]", "[/INST]"
     WIDTH                      = 50
     SIDE                       = (100 - WIDTH) / 2
     podcast_url_video_df = get_podcast_data(PODCAST_URL_VIDEO_PATH)
     together, translator, nlp, retriever, video_option, video_option_joined_path, default_system_prompt, youtube_video_url = get_basics_comp(EMB_MODEL, MODEL,
                                                                                                                                              DEFAULT_SYSTEM_PROMPT_LINK, logger,
                                                                                                                                              podcast_url_video_df, img_size=100)
     # -- 6. Setup prompt template + llm chain
     instruction = """CONTEXTO:/n/n {context}/n
     if "messages" not in st.session_state:
         st.session_state.messages = []
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
+            print("MUESTRO! " + message["content"])
             st.markdown(message["content"])
     if prompt := st.chat_input("¡Pregunta lo que quieras!"):
         with st.chat_message("user"):
             st.markdown(prompt)