Update app.py

app.py

@@ -55,7 +55,7 @@ class LazyPipeline:
                 "text-generation",
                 model=model,
                 tokenizer=tokenizer,
-                max_new_tokens=2048,
+                max_new_tokens=4096,
                 temperature=0.6,
             )
         return self.pipeline
@@ -181,7 +181,7 @@ def process_input(input_file, progress=gr.Progress()):
         with open(srt_path, 'r', encoding='utf-8') as file:
             content = file.read()
         words, tokens = count_words_and_tokens(content)
-        input_info = f"Video transcribed. Words: {words}, Tokens: {tokens}"
+        input_info = f"Input Words: {words} / Input Tokens: {tokens}"
     else:
         return "Unsupported file format. Please upload a TXT, PDF, or video file.", None, None, None, None, None, None
 
@@ -258,7 +258,7 @@ def update_visibility_and_charts(status, exec_time, lang, info, attachments, big
             gr.update(value=exec_time, visible=True),
             gr.update(value=lang, visible=True),
             gr.update(value=info, visible=True),
-        ] + [None for _ in range(12)]
+        ] + []  # No charts to return
 
     for analysis_text in [attachments, bigfive, personalities]:
         speakers_data = extract_speaker_data(analysis_text)
@@ -283,15 +283,12 @@ def update_visibility_and_charts(status, exec_time, lang, info, attachments, big
 
     print("Number of charts created:", len(charts))
 
-    # Ensure we always return the same number of outputs
-    chart_outputs = charts + [None] * (12 - len(charts))
-
     return [
         gr.update(value=status, visible=True),
         gr.update(value=exec_time, visible=True),
         gr.update(value=lang, visible=True),
         gr.update(value=info, visible=True),
-    ] + chart_outputs
+    ] + charts
 
 def create_interface():
     with gr.Blocks() as iface:
@@ -313,18 +310,26 @@ def create_interface():
             bigfive_output = gr.Textbox(visible=False)
             personalities_output = gr.Textbox(visible=False)
             
-            charts = [gr.Plot(visible=True) for _ in range(12)]  # Assuming maximum 12 charts
+            # Container for dynamically created charts
+            chart_container = gr.Column()
+
+        def process_and_update(input_file):
+            # First, process the input
+            results = process_input(input_file)
+            
+            # Then, create and update charts
+            chart_outputs = update_visibility_and_charts(*results)
+            
+            # Create new chart components based on the number of charts
+            new_charts = [gr.Plot(visible=True) for _ in range(len(chart_outputs) - 4)]  # -4 for the non-chart outputs
+            
+            # Update the chart container
+            return chart_outputs[:4] + [gr.Column(new_charts)]
 
         input_file.upload(
-            fn=process_input,
+            fn=process_and_update,
             inputs=[input_file],
-            outputs=[status_text, execution_time, detected_language, input_info, 
-                     attachments_output, bigfive_output, personalities_output]
-        ).then(
-            fn=update_visibility_and_charts,
-            inputs=[status_text, execution_time, detected_language, input_info, 
-                    attachments_output, bigfive_output, personalities_output],
-            outputs=[status_text, execution_time, detected_language, input_info] + charts
+            outputs=[status_text, execution_time, detected_language, input_info, chart_container]
         )
 
     return iface