Update app.py

app.py

@@ -90,13 +90,14 @@ def process_video(video_file, progress=gr.Progress()):
         progress(0.8, desc="Retrieving results...")
         
         # Retrieve transcription and insights
-        result = calls_table.select(calls_table.transcription_text, calls_table.insights).tail(1)
+        result = calls_table.select(calls_table.transcription_text, calls_table.insights, calls_table.audio).tail(1)
         transcription = result['transcription_text'][0]
         insights = result['insights'][0]
+        audio = result['audio'][0]
 
         progress(1.0, desc="Processing complete")
 
-        return transcription, insights, "Processing complete"
+        return transcription, insights, audio, "Processing complete"
 
     except Exception as e:
         return f"An error occurred during video processing: {str(e)}", ""
@@ -144,8 +145,8 @@ def chatbot_response(message, chat_history):
         temp_table['chatbot_prompt'] = create_chatbot_prompt(context_text, temp_table.question)
         temp_table['chatbot_response'] = openai.chat_completions(
             messages=temp_table.chatbot_prompt,
-            model='gpt-3.5-turbo',
-            max_tokens=150
+            model='gpt-4o-mini-2024-07-18',
+            max_tokens=300
         ) 
         temp_table['answer'] = temp_table.chatbot_response.choices[0].message.content
         
@@ -210,19 +211,23 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
         with gr.Column(scale=2):
             with gr.Tabs() as tabs:
                 with gr.TabItem("📝 Transcript"):
-                    output_transcription = gr.Textbox(label="Call Transcription", lines=15)
+                    output_transcription = gr.Textbox(label="Call Transcription", lines=10)
                 
                 with gr.TabItem("💡 Insights"):
-                    output_insights = gr.Textbox(label="Key Takeaways", lines=10)
+                    output_insights = gr.Textbox(label="Key Takeaways", lines=20)
+
+                with gr.TabItem("🎵 Audio"):
+                    output_audio = gr.Audio(label="Extracted Audio", type="filepath")
             
-                with gr.TabItem("🔍 Similarity Search"):
+                with gr.TabItem("🔍 Search"):
                     with gr.Row():
                         similarity_query = gr.Textbox(label="Search Query", placeholder="Enter a topic or phrase to search for")
                         num_results = gr.Slider(minimum=1, maximum=20, value=5, step=1, label="Number of Results")
                     similarity_search_btn = gr.Button("Search", variant="secondary")
                     similarity_results = gr.DataFrame(
                         headers=["Relevant Text", "Similarity Score"],
-                        label="Search Results"
+                        label="Search Results",
+                        wrap=True
                     )
             
                 with gr.TabItem("🤖 AI Assistant"):
@@ -246,7 +251,7 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
     process_btn.click(
         process_video,
         inputs=[video_file],
-        outputs=[output_transcription, output_insights, status_output],
+        outputs=[output_transcription, output_insights, output_audio, status_output],
         show_progress="full"
     )