Update app.py

app.py

@@ -12,6 +12,7 @@ import psutil
 
 st.set_page_config(layout="wide")
 
+# CSS Styling
 st.markdown("""
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600;700&display=swap');
@@ -58,7 +59,6 @@ st.markdown("""
     padding: 10px 15px;
     transition: text-shadow 0.3s ease-in-out;
     text-shadow: 5px 5px 12px rgba(0, 0, 0, 0.5);
-    
 }
 .navbar li a:hover {
     color: #ff6f61;
@@ -107,11 +107,10 @@ st.markdown("""
     box-shadow: 0 12px 24px rgba(0, 0, 0, 0.25);
     transition: all 0.3s ease;
     border: 1px solid rgba(0, 0, 0, 0.1);
-    background-color: #fff; /* or a slightly brighter color */
+    background-color: #fff;
     filter: brightness(1.05);
     z-index: 10;
 }
-
 .feature i {
     font-size: 1.5rem;
     color: #2196f3;
@@ -138,12 +137,10 @@ st.markdown("""
     text-align: center;
     box-shadow: 0 4px 10px rgba(0, 0, 0, 0.1);
     transition: transform 0.3s ease, box-shadow 0.3s ease;
-                border-top: 4px solid #28a745;
+    border-top: 4px solid #28a745;
     height: 290px;
     padding-top: 10px;
 }
-
-}
 .plan:hover {
     transform: translateY(-5px);
     box-shadow: 0 6px 15px rgba(0, 0, 0, 0.15);
@@ -152,7 +149,6 @@ st.markdown("""
     font-size: 1.5rem;
     margin-bottom: 0.5rem;
 }
-            
 .plan.free { border-top: 4px solid #28a745; }
 .plan.premium { border-top: 4px solid #ff6f61; }
 .plan.business { border-top: 4px solid #2196f3; }
@@ -194,6 +190,17 @@ st.markdown("""
     font-family: 'Poppins', sans-serif;
 }
 
+/* Video player styling */
+video {
+    display: block;
+    width: 350px !important;
+    height: 500px !important;
+    object-fit: contain;
+    margin: 0 auto;
+    border: 3px solid #2196f3;
+    border-radius: 8px;
+}
+
 /* Footer */
 footer {
     background: #1a1a1a;
@@ -255,6 +262,7 @@ footer {
 <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0-beta3/css/all.min.css">
 """, unsafe_allow_html=True)
 
+# Function Definitions
 def format_time(seconds):
     minutes = int(seconds // 60)
     secs = int(seconds % 60)
@@ -285,8 +293,8 @@ class TranscriptionProgress:
 def load_model(language='en', summarizer_type='bart'):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     if language == 'ur':
-        processor = AutoProcessor.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs")
-        model = AutoModelForSpeechSeq2Seq.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs").to(device)
+        processor = AutoProcessor.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs-v2")
+        model = AutoModelForSpeechSeq2Seq.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs-v2").to(device)
     else:
         processor = AutoProcessor.from_pretrained("openai/whisper-small")
         model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-small").to(device)
@@ -449,6 +457,7 @@ def generate_srt(transcript, include_timeframe=True):
             srt_content += f"{text}\n\n"
     return srt_content
 
+# Main Function
 def main():
     st.markdown("""
     <div class="header">
@@ -471,33 +480,7 @@ def main():
     </div>
     """, unsafe_allow_html=True)
 
-    st.markdown("""
-    <style>
-    /* Video player styling */
-    .stVideo > div {
-        max-width: 600px !important;
-        max-height: 450px !important;
-        margin: 0 auto;
-        border: 3px solid #2196f3;
-        border-radius: 8px;
-        overflow: hidden;
-    }
-    
-    /* Footer column styling */
-    .footer-container {
-        display: flex;
-        justify-content: space-between;
-    }
-    .footer-section:first-child {
-        margin-left: 10px;
-    }
-    .footer-section:last-child {
-        margin-right: 10px;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-
+    # Initialize session state
     if 'app_state' not in st.session_state:
         st.session_state['app_state'] = 'upload'
     if 'video_path' not in st.session_state:
@@ -539,8 +522,6 @@ def main():
                     st.session_state['app_state'] = 'processing'
                     st.write(f"Uploaded file: {uploaded_file.name}")
                     st.rerun()
-    
-    
 
     if st.session_state['app_state'] == 'processing':
         with st.form(key="processing_form"):
@@ -601,28 +582,50 @@ def main():
                                 os.remove(temp_file)
 
     if st.session_state['app_state'] == 'results':
+        st.markdown('<div style="display: flex; justify-content: center;">', unsafe_allow_html=True)
         st.video(st.session_state['video_path'], start_time=st.session_state['current_time'])
+        st.markdown('</div>', unsafe_allow_html=True)
+        
         st.session_state['show_timeframe'] = st.checkbox("Show timeframe in transcript", value=st.session_state['show_timeframe'])
         st.markdown("### Search Subtitles")
-        search_query = st.text_input("Search subtitles...", value=st.session_state['search_query'], key="search_input")
-        st.session_state['search_query'] = search_query.lower()
+        
+        # Callback to handle search query updates
+        def update_search_query():
+            st.session_state['search_query'] = st.session_state.get('search_input', '').lower().strip()
+        
+        # Text input with on_change callback
+        st.text_input("Search subtitles...", value=st.session_state['search_query'], key="search_input", on_change=update_search_query)
+
+        # Primary Transcript
         st.markdown(f"### {st.session_state['language']} Transcript")
+        primary_matches = 0
         for text, start, end in st.session_state['primary_transcript']:
-            display_text = text.lower()
-            if not search_query or search_query in display_text:
+            display_text = text.lower()  # Case-insensitive comparison
+            if not st.session_state['search_query'] or st.session_state['search_query'] in display_text:
+                primary_matches += 1
                 label = f"[{format_time(start)} - {format_time(end)}] {text}" if st.session_state['show_timeframe'] else text
                 if st.button(label, key=f"primary_{start}"):
                     st.session_state['current_time'] = start
                     st.rerun()
+        if primary_matches == 0 and st.session_state['search_query']:
+            st.info("No matches found in primary transcript for the search query.")
+
+        # English Transcript
         if st.session_state['english_transcript']:
             st.markdown("### English Translation")
+            english_matches = 0
             for text, start, end in st.session_state['english_transcript']:
-                display_text = text.lower()
-                if not search_query or search_query in display_text:
+                display_text = text.lower()  # Case-insensitive comparison
+                if not st.session_state['search_query'] or st.session_state['search_query'] in display_text:
+                    english_matches += 1
                     label = f"[{format_time(start)} - {format_time(end)}] {text}" if st.session_state['show_timeframe'] else text
                     if st.button(label, key=f"english_{start}"):
                         st.session_state['current_time'] = start
                         st.rerun()
+            if english_matches == 0 and st.session_state['search_query']:
+                st.info("No matches found in English transcript for the search query.")
+
+        # Summary Generation
         if (st.session_state['language_code'] == 'en' or st.session_state['translate_to_english']) and not st.session_state['summary_generated']:
             if st.button("Generate Summary"):
                 with st.spinner("Generating summary..."):
@@ -637,12 +640,16 @@ def main():
         if st.session_state['english_summary'] and st.session_state['summary_generated']:
             st.markdown("### Summary")
             st.write(st.session_state['english_summary'])
+
+        # Download Subtitles
         st.markdown("### Download Subtitles")
         include_timeframe = st.checkbox("Include timeframe in subtitles", value=True)
         transcript_to_download = st.session_state['primary_transcript'] or st.session_state['english_transcript']
         if transcript_to_download:
             srt_content = generate_srt(transcript_to_download, include_timeframe)
             st.download_button(label="Download Subtitles (SRT)", data=srt_content, file_name="subtitles.srt", mime="text/plain")
+
+        # Edit Subtitles
         st.markdown("### Edit Subtitles")
         transcript_to_edit = st.session_state['primary_transcript'] or st.session_state['english_transcript']
         if transcript_to_edit and st.button("Delete Subtitles"):
@@ -678,7 +685,9 @@ def main():
 
     if st.session_state['app_state'] == 'results' and st.session_state['edited_video_path']:
         st.markdown("### Edited Video")
+        st.markdown('<div style="display: flex; justify-content: center;">', unsafe_allow_html=True)
         st.video(st.session_state['edited_video_path'])
+        st.markdown('</div>', unsafe_allow_html=True)
         with open(st.session_state['edited_video_path'], "rb") as file:
             st.download_button(label="Download Edited Video", data=file, file_name="edited_video.mp4", mime="video/mp4")
 
@@ -703,29 +712,29 @@ def main():
     """, unsafe_allow_html=True)
 
     st.markdown("""
-        <div id="about" class="about-section" style="padding: 3rem 2rem; background: #f8f9fa; border-radius: 1rem; margin: 2rem 0;">
-            <h2 style="text-align: center; color: black; margin-bottom: 2rem;">About VidEp</h2>
-            <div style="display: flex; align-items: center; gap: 2rem; flex-wrap: wrap;">
-                <div style="flex: 1; min-width: 300px;">
-                    <img src="https://i.postimg.cc/g0z3WVgT/about.jpg" style="width: 100%; height: auto; border-radius: 1rem;" alt="About VidEp">
-                </div>
-                <div style="flex: 2; min-width: 300px;">
-                    <h3 style="color:grey;">Our Mission</h3>
-                    <p>VidEp aims to revolutionize how creators and professionals work with video content by providing state-of-the-art AI-powered tools for transcription, translation, and summarization.</p>
-                    <h3 style="color:grey;">What We Do</h3>
-                    <p>Our platform combines the latest advancements in speech recognition and natural language processing to automatically transcribe videos in multiple languages, generate accurate translations, and create concise summaries of content.</p>
-                    <h3 style="color:grey;">Why Choose Us</h3>
-                    <ul>
-                        <li>Advanced AI models for superior accuracy</li>
-                        <li>Multi-language support including English and Urdu</li>
-                        <li>Easy-to-use interface for editing and managing subtitles</li>
-                        <li>Smart search functionality to quickly find content</li>
-                        <li>Seamless video editing based on transcripts</li>
-                    </ul>
-                </div>
+    <div id="about" class="about-section" style="padding: 3rem 2rem; background: #f8f9fa; border-radius: 1rem; margin: 2rem 0;">
+        <h2 style="text-align: center; color: black; margin-bottom: 2rem;">About VidEp</h2>
+        <div style="display: flex; align-items: center; gap: 2rem; flex-wrap: wrap;">
+            <div style="flex: 1; min-width: 300px;">
+                <img src="https://i.postimg.cc/g0z3WVgT/about.jpg" style="width: 100%; height: auto; border-radius: 1rem;" alt="About VidEp">
+            </div>
+            <div style="flex: 2; min-width: 300px;">
+                <h3 style="color:grey;">Our Mission</h3>
+                <p>VidEp aims to revolutionize how creators and professionals work with video content by providing state-of-the-art AI-powered tools for transcription, translation, and summarization.</p>
+                <h3 style="color:grey;">What We Do</h3>
+                <p>Our platform combines the latest advancements in speech recognition and natural language processing to automatically transcribe videos in multiple languages, generate accurate translations, and create concise summaries of content.</p>
+                <h3 style="color:grey;">Why Choose Us</h3>
+                <ul>
+                    <li>Advanced AI models for superior accuracy</li>
+                    <li>Multi-language support including English and Urdu</li>
+                    <li>Easy-to-use interface for editing and managing subtitles</li>
+                    <li>Smart search functionality to quickly find content</li>
+                    <li>Seamless video editing based on transcripts</li>
+                </ul>
             </div>
         </div>
-        """, unsafe_allow_html=True)
+    </div>
+    """, unsafe_allow_html=True)
 
     st.markdown("""
     <div id="contact" class="contact-section" style="padding: 3rem 2rem; background: #f1f4f8; border-radius: 1rem; margin: 2rem 0;">
@@ -733,11 +742,11 @@ def main():
         <div style="max-width: 600px; margin: 0 auto;">
             <div style="margin-bottom: 1rem;">
                 <label for="email" style="display: block; margin-bottom: 0.5rem; font-weight: 500;">Email</label>
-                <input type="email" id="email" placeholder="Your email address" style="width: 100%; padding: 0.75rem; background-color:white; color:white; border-radius: 0.5rem; border: 1px solid #e0e0e0;">
+                <input type="email" id="email" placeholder="Your email address" style="width: 100%; padding: 0.75rem; border-radius: 0.5rem; border: 1px solid #e0e0e0;">
             </div>
             <div style="margin-bottom: 1rem;">
                 <label for="message" style="display: block; margin-bottom: 0.5rem; font-weight: 500;">Message</label>
-                <textarea id="message" rows="5" placeholder="Your message" style="width: 100%; padding: 0.75rem; border-radius: 0.5rem; background-color:white; color:white; border: 1px solid #e0e0e0;"></textarea>
+                <textarea id="message" rows="5" placeholder="Your message" style="width: 100%; padding: 0.75rem; border-radius: 0.5rem; border: 1px solid #e0e0e0;"></textarea>
             </div>
             <button onclick="alert('Message sent successfully!')" style="background: linear-gradient(135deg, #ff6f61, #ff8a65); color: white; font-weight: 600; padding: 0.75rem 1.5rem; border-radius: 0.5rem; border: none; cursor: pointer; width: 100%;">Send Message</button>
         </div>
@@ -746,11 +755,9 @@ def main():
 
     st.markdown("""
     <div class="plans">
-        <h2 style="text-align: center; margin-bottom: 2rem;color: black;">Choose Your Plan</h2>
+        <h2 style="text-align: center; margin-bottom: 2rem; color: black;">Choose Your Plan</h2>
         <div class="plan-box">
-            <div class="plan free" 
-                style="background: linear-gradient(135deg, #299f45, #185726);
-                 padding-bottom: 0px">
+            <div class="plan free" style="background: linear-gradient(135deg, #299f45, #185726); padding-bottom: 0px">
                 <h3 style="color: white;">Free</h3>
                 <p><strong>$0</strong> / month</p>
                 <p>Basic video transcription</p>
@@ -758,7 +765,7 @@ def main():
                 <p>Max 5 minutes video</p>
                 <p>No summarization</p>
             </div>
-            <div class="plan premium"  style="background-color:#a32b2d">
+            <div class="plan premium" style="background-color:#a32b2d">
                 <h3 style="color: white;">Premium</h3>
                 <p><strong>$19</strong> / month</p>
                 <p>Advanced transcription</p>
@@ -766,7 +773,7 @@ def main():
                 <p>Max 30 minutes video</p>
                 <p>AI summarization</p>
             </div>
-            <div class="plan business"  style="background-color:#396ca3">
+            <div class="plan business" style="background-color:#396ca3">
                 <h3 style="color: white;">Business</h3>
                 <p><strong>$49</strong> / month</p>
                 <p>Enterprise-grade transcription</p>
@@ -827,5 +834,6 @@ def main():
     });
     </script>
     """, unsafe_allow_html=True)
+
 if __name__ == "__main__":
     main()