percentage progress

app.py

@@ -72,23 +72,30 @@ def transcribe_audio(audio_file):
     return transcription
 
 @spaces.GPU
-def analyze_video(prompt, video):
+def analyze_video(prompt, video, progress=gr.Progress()):
     start_time = time.time()
     
+    progress(0, desc="Initializing")
+    
     if isinstance(video, str):
         video_path = video
     else:
         video_path = video.name
     
+    progress(10, desc="Encoding video")
     encoded_video = encode_video(video_path)
     
+    progress(30, desc="Extracting audio")
     # Extract audio and transcribe
     audio_path = extract_audio(video_path)
+    
+    progress(50, desc="Transcribing audio")
     transcription = transcribe_audio(audio_path)
     
     # Clean up temporary audio file
     os.remove(audio_path)
     
+    progress(70, desc="Preparing context")
     context = [
         {"role": "user", "content": [prompt] + encoded_video},
         {"role": "assistant", "content": f"Transcription of the video: {transcription}"}
@@ -105,12 +112,16 @@ def analyze_video(prompt, video):
         "max_slice_nums": 1 if len(encoded_video) > 16 else 2
     }
     
+    progress(80, desc="Generating response")
     response = model.chat(image=None, msgs=context, tokenizer=tokenizer, **params)
     
+    progress(90, desc="Finalizing")
     end_time = time.time()
     processing_time = end_time - start_time
     analysis_result = f"Analysis Result:\n{response}\n\n"
     processing_time = f"Processing Time: {processing_time:.2f} seconds"
+    
+    progress(100, desc="Complete")
     return analysis_result, processing_time
 
 with gr.Blocks() as demo: