Update app.py

app.py

@@ -17,15 +17,26 @@ class ConsoleOutput:
     def get_output(self):
         return "\n".join(self.messages)
 
+class GradioHandler(logging.Handler):
+    def emit(self, record):
+        msg = self.format(record)
+        self.console.write(msg)
+
+def setup_logging(console_output):
+    logging.basicConfig(level=logging.DEBUG)
+    logger = logging.getLogger()
+    logger.handlers = []
+    handler = GradioHandler()
+    handler.console = console_output
+    logger.addHandler(handler)
+    return logger
+
 def process_screenplay(pdf_file, progress=gr.Progress()):
     if pdf_file is None:
         raise gr.Error("Please upload a PDF file")
     
     console = ConsoleOutput()
-    logging.basicConfig(level=logging.DEBUG)
-    logger = logging.getLogger()
-    logger.handlers = []
-    logger.addHandler(GradioHandler())
+    logger = setup_logging(console)
     
     try:
         processor = GeminiProcessor()
@@ -50,10 +61,7 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
 
 def generate_coverage(progress=gr.Progress()):
     console = ConsoleOutput()
-    logging.basicConfig(level=logging.DEBUG)
-    logger = logging.getLogger()
-    logger.handlers = []
-    logger.addHandler(GradioHandler())
+    logger = setup_logging(console)
     
     try:
         coverage_gen = CoverageGenerator()
@@ -77,10 +85,7 @@ def generate_coverage(progress=gr.Progress()):
 
 def analyze_screenplay(progress=gr.Progress()):
     console = ConsoleOutput()
-    logging.basicConfig(level=logging.DEBUG)
-    logger = logging.getLogger()
-    logger.handlers = []
-    logger.addHandler(GradioHandler())
+    logger = setup_logging(console)
     
     try:
         analyzer = CreativeAnalyzer()