Update app.py

app.py

@@ -18,9 +18,8 @@ from groq import Groq
 
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-TRANSCRIPTION_MODEL = "turbowhisper-large-v3"   # updated model
-GRAMMAR_MODEL = "openai/gpt-oss-120b"           # grammar & punctuation fix
-
+TRANSCRIPTION_MODEL = "turbowhisper-large-v3"
+REWRITE_MODEL = "openai/gpt-oss-120b"
 CHUNK_SIZE = 600  # 10 min in seconds
 
 # ============================================================================
@@ -28,7 +27,7 @@ CHUNK_SIZE = 600  # 10 min in seconds
 # ============================================================================
 
 def get_video_duration(video_path):
-    """Get video duration using ffprobe (more reliable than imageio)"""
+    """Get video duration using ffprobe"""
     try:
         result = subprocess.run(
             ["ffprobe", "-v", "error", "-show_entries",
@@ -100,41 +99,46 @@ def transcribe_audio_chunks(audio_chunks, language="English"):
     return transcript_text.strip()
 
 # ============================================================================
-# GRAMMAR & PUNCTUATION FIX
+# GRAMMAR & PUNCTUATION CORRECTION
 # ============================================================================
 
-def correct_grammar(text, language="English"):
-    """Use Groq GPT model to fix grammar and punctuation"""
+def rewrite_transcript(text, language="English"):
+    """Rewrite transcript using GPT-OSS-120B with correct grammar & punctuation"""
     try:
-        prompt = f"Correct the grammar and punctuation for the following text in {language}:\n\n{text}"
+        prompt = f"Correct grammar, punctuation, and make it readable. Language: {language}\n\n{text}"
         response = client.chat.completions.create(
-            model=GRAMMAR_MODEL,
-            messages=[{"role": "user", "content": prompt}],
+            model=REWRITE_MODEL,
+            messages=[{"role":"user", "content": prompt}],
             temperature=0
         )
-        fixed_text = response.choices[0].message.content
-        return fixed_text
+        rewritten_text = response.choices[0].message.content
+        return rewritten_text
     except Exception as e:
-        return f"[Grammar correction failed: {str(e)}]"
+        return f"[Error rewriting transcript: {str(e)}]"
 
 # ============================================================================
 # PROCESS VIDEO
 # ============================================================================
 
-def process_video_ui(video_file, language):
+def generate_transcript(video_file, language):
     lang = "English" if "English" in language else "Urdu"
     if not video_file:
         return "", "Please upload a video file." if lang=="English" else "براہ کرم ایک ویڈیو فائل اپ لوڈ کریں۔"
     try:
         audio_chunks = extract_audio_chunks(video_file.name)
         transcript = transcribe_audio_chunks(audio_chunks, language=lang)
-        # fix grammar & punctuation
-        transcript_fixed = correct_grammar(transcript, language=lang)
-        return transcript_fixed, "✅ Success / کامیاب"
+        return transcript, "✅ Transcription completed / ٹرانسکرپشن مکمل"
     except Exception as e:
         error_msg = str(e)
         return "", error_msg if lang=="English" else "خرابی: " + error_msg
 
+def correct_transcript(transcript_text, language):
+    lang = "English" if "English" in language else "Urdu"
+    if not transcript_text:
+        return "", "Please generate transcript first." if lang=="English" else "پہلے ٹرانسکرپٹ بنائیں۔"
+    rewritten_text = rewrite_transcript(transcript_text, language=lang)
+    return rewritten_text, "✅ Grammar & punctuation corrected / گرامر اور پنکچویشن درست کی گئی"
+
 # ============================================================================
 # GRADIO UI
 # ============================================================================
@@ -142,17 +146,30 @@ def process_video_ui(video_file, language):
 def create_ui():
     with gr.Blocks(title="AI Graduate Job Matcher") as demo:
         gr.HTML("<h1>🎓 AI Graduate Job-Matching Platform</h1><p>Developer: Najaf Ali Sharqi</p>")
+        
         video_input = gr.File(label="Upload Video / ویڈیو اپ لوڈ کریں", file_types=[".mp4", ".mov", ".avi"])
         language_dropdown = gr.Dropdown(choices=["English", "Urdu / اردو"], value="English", label="Output Language / آؤٹ پٹ زبان")
-        process_btn = gr.Button("🚀 Generate Transcript / ٹرانسکرپٹ بنائیں")
+        
         transcript_output = gr.Textbox(label="Transcript / ٹرانسکرپٹ", lines=20)
         status_output = gr.Textbox(label="Status / حیثیت", lines=1)
         
-        process_btn.click(
-            fn=process_video_ui,
+        # Buttons
+        transcribe_btn = gr.Button("🚀 Generate Transcript / ٹرانسکرپٹ بنائیں")
+        rewrite_btn = gr.Button("✏️ Correct Grammar & Punctuation / گرامر درست کریں")
+        
+        # Transcription click
+        transcribe_btn.click(
+            fn=generate_transcript,
             inputs=[video_input, language_dropdown],
             outputs=[transcript_output, status_output]
         )
+        
+        # Grammar correction click
+        rewrite_btn.click(
+            fn=correct_transcript,
+            inputs=[transcript_output, language_dropdown],
+            outputs=[transcript_output, status_output]
+        )
     
     return demo