Upload 3 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+header-image-2.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -6,6 +6,77 @@ import requests
 import streamlit as st
 from PIL import Image
 
+def inject_custom_css():
+    st.markdown(
+        """
+        <style>
+        @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700&display=swap');
+        body {
+            font-family: 'Roboto', sans-serif;
+            background-color: #f5f5f5;
+            color: #333;
+        }
+        body {
+            padding-top: 0px;
+        }
+        .stApp {
+            padding-top: 10px;
+        }
+        .stButton>button {
+            background-color: #4CAF50;
+            color: white;
+            padding: 10px 24px;
+            border: none;
+            cursor: pointer;
+            border-radius: 4px;
+        }
+        .stButton>button:hover {
+            background-color: #45a049;
+        }
+        .stTextInput>div>input {
+            padding: 10px;
+            border-radius: 4px;
+            border: 1px solid #ccc;
+            font-size: 16px;
+        }
+        .stTextInput>div>input:focus {
+            border-color: #007BFF;
+            outline: none;
+            box-shadow: 0 0 5px rgba(0, 123, 255, 0.5);
+        }
+        .centered {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+        }
+        .sidebar .stImage {
+            display: flex;
+            justify-content: center;
+            margin-bottom: 20px;
+        }
+        .css-1lcbmhc.e1fqkh3o3 { /* This targets the sidebar */
+            background-color: #ffffff !important;
+            color: #333 !important;
+        }
+        .css-1d391kg { /* This targets the sidebar headings */
+            color: #333 !important;
+        }
+        .form-container {
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+        }
+        .form-container .stTextInput {
+            flex: 1;
+        }
+        .form-container .stButton {
+            margin-left: 10px;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )    
+
 def launch_bot():
     
     if 'cfg' not in st.session_state:
@@ -22,12 +93,18 @@ def launch_bot():
     cfg = st.session_state.cfg
     vq = st.session_state.vq
     st.set_page_config(page_title="Media Demo", layout="wide")
+    inject_custom_css()
+
+    header_image = Image.open('header-image-2.png')
+    cropped_image = header_image.crop((0, 0, header_image.width, 200))
+    st.image(cropped_image, use_column_width=True)
 
     # left side content
     with st.sidebar:
-        image = Image.open('Vectara-logo.png')
+        image = Image.open('vectara-logo.png')
         st.markdown("## Welcome to Media Demo\n\n"
-                    "This demo uses Vectara to find the movie where a quote is from\n\n")
+                    "This demo uses Vectara to find the movie where a quote is from\n\n"
+                    "Covers movies from this [playlist](https://www.youtube.com/playlist?list=PLHPTxTxtC0ibVZrT2_WKWUl2SAxsKuKwx) of free movies")
 
         st.markdown("---")
         st.markdown(
@@ -37,18 +114,26 @@ def launch_bot():
         st.markdown("---")
         st.image(image, width=250)
 
-    st.markdown("<center> <h2> Vectara Media Demo</h2> </center>", unsafe_allow_html=True)
-    
-    question = st.text_input("Enter your question:")
-    if st.button("find the match"):
-        movie_name, match_url = vq.submit_query(question)
-        video_url, start_time = match_url.split('&t=')
-        start_time = start_time[:-1]   # remove the trailing 's'
-
-        col1, col2, col3 = st.columns([1, 2, 1])
-        with col2:
-            st.write(f"Here's a useful video for you: {movie_name}")
-            st.video(video_url, start_time=int(float(start_time)))
+    st.markdown("<center> <h3>\"Find that movie\" demo</h3> </center>", unsafe_allow_html=True)
+
+    st.markdown('<div class="form-container">', unsafe_allow_html=True)
+    with st.form(key='my_form'):
+        question = st.text_input("Enter your movie quote:")
+        submit_button = st.form_submit_button(label='Find the Match')
+    st.markdown('</div>', unsafe_allow_html=True)
+
+    if submit_button:
+        movie_name, match_url, score = vq.submit_query(question)
+        if score < 0.7:
+            st.write("Sorry, I couldn't find a match for that quote. Please try another one.")
+        else:
+            video_url, start_time = match_url.split('&t=')
+            start_time = start_time[:-1]   # remove the trailing 's'
+
+            col1, col2, col3 = st.columns([1, 2, 1])
+            with col2:
+                st.write(f"Here's a useful video for you: {movie_name}")
+                st.video(video_url, start_time=int(float(start_time)))
 
 if __name__ == "__main__":
     launch_bot()

query.py

@@ -27,10 +27,7 @@ class VectaraQuery():
                         'start_tag': "%START_SNIPPET%",
                         'end_tag': "%END_SNIPPET%",
                     },
-                    'rerankingConfig':
-                    {
-                        'rerankerId': 272725719,
-                    },
+                    'rerankingConfig': { 'rerankerId': 272725719 }
                 } 
             ]
         }
@@ -59,6 +56,7 @@ class VectaraQuery():
         responses = res['responseSet'][0]['response'][:top_k]
         documents = res['responseSet'][0]['document']
 
+
         metadatas = []
         for x in responses:
             md = {m["name"]: m["value"] for m in x["metadata"]}
@@ -69,6 +67,7 @@ class VectaraQuery():
 
         movie_title = metadatas[0].get("doc_title", None)
         snippet_url = metadatas[0].get("url", None)
+        score = responses[0]["score"]
 
-        return movie_title, snippet_url
+        return movie_title, snippet_url, score