Update src/processor.py

src/processor.py

@@ -6,7 +6,7 @@ import requests
 import tempfile
 import gradio as gr
 
-def process_document(pdf_file, api_key, voice_choice, style_choice, chunk_size, temperature, max_tokens, progress=gr.Progress()):
+def process_document(pdf_file, api_key, voice_choice, style_choice, chunk_size, temperature, max_tokens, system_prompt, progress=gr.Progress()):
     try:
         os.environ['OPENAI_API_KEY'] = api_key
         
@@ -19,12 +19,18 @@ def process_document(pdf_file, api_key, voice_choice, style_choice, chunk_size,
             {
                 'document': pxt.Document,
                 'voice': pxt.String,
-                'style': pxt.String
+                'style': pxt.String,
+                'mode_prompt': pxt.String
             }
         )
         
         progress(0.2, desc="Processing document...")
-        docs.insert([{'document': pdf_file.name, 'voice': voice_choice, 'style': style_choice}])
+        docs.insert([{
+            'document': pdf_file.name,
+            'voice': voice_choice,
+            'style': style_choice,
+            'mode_prompt': system_prompt
+        }])
         
         chunks = pxt.create_view(
             'document_audio.chunks',
@@ -41,11 +47,7 @@ def process_document(pdf_file, api_key, voice_choice, style_choice, chunk_size,
             messages=[
                 {
                     'role': 'system',
-                    'content': """Transform this text segment into clear, concise content.
-                    Structure:
-                    1. Core concepts and points
-                    2. Supporting details
-                    3. Key takeaways"""
+                    'content': docs.mode_prompt  # Use the mode-specific prompt
                 },
                 {'role': 'user', 'content': chunks.text}
             ],
@@ -53,6 +55,7 @@ def process_document(pdf_file, api_key, voice_choice, style_choice, chunk_size,
             max_tokens=max_tokens,
             temperature=temperature
         )
+        
         chunks['content'] = chunks.content_response['choices'][0]['message']['content']
 
         progress(0.6, desc="Script generation...")