Update app.py

app.py

@@ -26,18 +26,18 @@ def generate_lesson(doc_text=None, pdf_file=None):
             logger.info(f"Processing uploaded PDF file: {pdf_file.name}")
             doc_text = pdf_to_text(pdf_file.name)
 
-        logger.info("Processing the document text.")
-        processed_text = process_large_text(doc_text)
-        
-        logger.info("Generating lesson from transcript.")
-        generated_text, output_path = generate_lesson_from_transcript_logic(processed_text)
-
-        if output_path:
-            logger.info(f"Lesson generated successfully. Output saved to: {output_path}")
-            return generated_text, gr.File(output_path)
-        else:
-            logger.info("Lesson generated successfully, but no output file created.")
-            return generated_text, None
+        logger.info("Processing the document text with general model.")
+        general_model_output = process_large_text(doc_text)
+
+        logger.info("Refining the output with fine-tuned model.")
+        refined_output = refine_with_fine_tuned_model(general_model_output)
+
+        output_path = "/tmp/refined_output.txt"
+        with open(output_path, "w") as file:
+            file.write(refined_output)
+
+        logger.info(f"Lesson generated successfully. Output saved to: {output_path}")
+        return refined_output, gr.File(output_path)
 
     except Exception as e:
         logger.error(f"Error occurred while generating lesson: {str(e)}")