Update app.py

app.py

@@ -23,7 +23,6 @@ def transcribe_audio(audio_filepath):
         )
     return transcription.text
 
-
 def extract_text_from_pdf(pdf_path):
     text = ''
     with fitz.open(pdf_path) as doc:
@@ -31,8 +30,14 @@ def extract_text_from_pdf(pdf_path):
             text += page.get_text()
     return text
 
+def extract_texts_from_pdfs(pdfs):
+    text = ''
+    for pdf in pdfs:
+        pdf_text = extract_text_from_pdf(pdf.name)
+        text += pdf_text + "\n"
+    return text
+
 def organize_clinical_record(current_text, transcription_text, pdf_text):
-    # No cambiamos el clinical_record_template
     clinical_record_template = """
     MOTIVO DE CONSULTA: usa una frase en palabras del paciente entre comillas
 
@@ -70,7 +75,7 @@ def organize_clinical_record(current_text, transcription_text, pdf_text):
     {clinical_record_template}
 
     Borrador Actual del Registro Clínico:
-    {iterative_output}
+    {current_text}
 
     Nueva Información de Audio:
     {transcription_text}
@@ -84,14 +89,13 @@ def organize_clinical_record(current_text, transcription_text, pdf_text):
     organized_text = chat_groq.invoke(prompt)
     return organized_text
 
-def process_input(audio, pdfs):
+def process_input(audio, pdfs, current_text):
     transcription_text = transcribe_audio(audio)
     pdf_text = ''
     if pdfs:
         pdf_text = extract_texts_from_pdfs(pdfs)
-    combined_text = transcription_text + "\n" + pdf_text
-    organized_record = organize_clinical_record(combined_text)
-    return organized_record.content
+    organized_record = organize_clinical_record(current_text, transcription_text, pdf_text)
+    return organized_record.content, "Información de depuración"
 
 theme = gr.themes.Base(
     primary_hue=gr.themes.Color(
@@ -104,7 +108,7 @@ theme = gr.themes.Base(
     neutral_hue="neutral",
 )
 
-iterative_output = """
+initial_text = """
         MOTIVO DE CONSULTA: 
         
         ENFERMEDAD ACTUAL:
@@ -132,25 +136,21 @@ with gr.Blocks(theme=theme) as iface:
     
     iterative_output = gr.Textbox(
         label="Registro Clínico Organizado",
-        value= iterative_output,
+        value=initial_text,
         lines=20,
-            )
-    # Move the State inside the Blocks context
-    current_state = gr.State(value=iterative_output)
+    )
+    current_state = gr.State(value=initial_text)
     audio_filepath = gr.Audio(sources=["microphone"], type="filepath", label="Entrada de Audio")
     pdf_files = gr.File(file_types=[".pdf"], label="Subir PDFs (puedes subir múltiples archivos)", file_count="multiple")
     debug_output = gr.Textbox(label="Información de Depuración", lines=10)
     
     def on_audio_change(audio_filepath, pdfs, current_text):
         print(f"on_audio_change: audio_filepath = {audio_filepath}")
-        # Use the current_text passed in (which is current_state)
         if not current_text or not current_text.strip():
-            current_text = iterative_output.value  # Initial value if current_text is empty
+            current_text = initial_text
         updated_text, debug_info = process_input(audio_filepath, pdfs, current_text)
-        # Return updated_text to update both iterative_output and current_state
-        return updated_text, debug_info, updated_text  # Note: Return updated_text for current_state
-
-    # Update outputs to include current_state
+        return updated_text, debug_info, updated_text
+    
     audio_filepath.change(
         fn=on_audio_change,
         inputs=[audio_filepath, pdf_files, current_state],
@@ -160,14 +160,14 @@ with gr.Blocks(theme=theme) as iface:
     def on_pdfs_change(audio_filepath, pdfs, current_text):
         print(f"on_pdfs_change: pdfs = {pdfs}")
         if not current_text or not current_text.strip():
-            current_text = iterative_output.value
+            current_text = initial_text
         updated_text, debug_info = process_input(audio_filepath, pdfs, current_text)
-        return updated_text, debug_info, updated_text  # Return updated_text for current_state
-
+        return updated_text, debug_info, updated_text
+    
     pdf_files.change(
         fn=on_pdfs_change,
         inputs=[audio_filepath, pdf_files, current_state],
         outputs=[iterative_output, debug_output, current_state]
     )
 
-iface.launch(auth=[("her", "her")])
+iface.launch(auth=[("her", "her")])