Add options to add video ID

model.py

@@ -7,6 +7,7 @@ from transformers import (
     pipeline,
     GenerationConfig
 )
+from textwrap import dedent
 
 class lamini:
     def __init__(self):
@@ -114,15 +115,67 @@ class qa_template:
             _type_: _description_
         """
         import gradio as gr
+        load = self.load
         
         def interface(msg, history):
             res = self.qa_inf.run(msg)
-            return str(res)
-        
-        ui = gr.ChatInterface(
-            fn=interface,
-            examples=["What is the video about?", "key points of the video"],
-            title=f"Question Mode - {title}",
-        )
-        
-        ui.launch()
+            history.append((msg, res))
+            return "", history
+        
+        def reload(video_id):
+            from utils import getSubsText
+            print(f"Setting up {video_id}")
+            subs = getSubsText(video_id)
+            _ = load(subs)
+        
+        with gr.Blocks() as demo:
+            with gr.Column():
+                gr.Markdown(dedent(f"""
+                    # video to QA
+                    A test implementation to use vectorstres and mini llms to create
+                    a question answer chatbot interface for _youtube videos_
+                                   """))
+                chatbot = gr.Chatbot()
+                with gr.Row():
+                    with gr.Column():
+                        videoId = gr.Textbox(label="Video ID", placeholder="Enter video ID here")
+                        msg = gr.Textbox(label="Question Box" , placeholder="Enter your question here")
+                    clear = gr.ClearButton([msg, videoId, chatbot])
+                
+                gr.Markdown(
+                    dedent("""
+                    
+                    ## Getting started
+                    to start up you need to enter the video ID of youtube video first
+                    
+                    Get a youtube video which has English dialog
+                    > ex: https://www.youtube.com/watch?v=BsnCpESUEqM
+                    
+                    in this `BsnCpESUEqM` is the video ID
+                    
+                    ```
+                    https://www.youtube.com/watch?v=BsnCpESUEqM
+                                                    ^^^^^^^^^^^
+                                                    video_id
+                    ```
+                    > in url paramets are seperated by `?` and for video id its `?v`
+                    
+                    copy-paste the video id to the textbox and press return/enter and wait ~5 seconds to fetch video information
+                    
+                    ---
+                    
+                    Now in the Question Box _box_/feild start typing the quesions and press return/enter to send to llm
+                    """)
+                )
+            msg.submit(interface, [msg, chatbot], [msg, chatbot])
+            videoId.submit(reload, [videoId])
+            
+        
+        # ui = gr.ChatInterface(
+        #     fn=interface,
+        #     examples=["What is the video about?", "key points of the video"],
+        #     title=f"Question Mode - {title}",
+        # )
+        
+        # ui.launch()
+        demo.launch()