Update app.py
Browse files
app.py
CHANGED
|
@@ -21,10 +21,10 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
|
|
| 21 |
raise gr.Error("Please upload a PDF file")
|
| 22 |
|
| 23 |
console = ConsoleOutput()
|
| 24 |
-
logging.basicConfig(level=logging.DEBUG
|
| 25 |
logger = logging.getLogger()
|
| 26 |
logger.handlers = []
|
| 27 |
-
|
| 28 |
class GradioHandler(logging.Handler):
|
| 29 |
def emit(self, record):
|
| 30 |
msg = self.format(record)
|
|
@@ -45,7 +45,7 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
|
|
| 45 |
cleaned_text = f.read()
|
| 46 |
|
| 47 |
progress(1.0, desc="Complete!")
|
| 48 |
-
return cleaned_text, True, console.get_output()
|
| 49 |
|
| 50 |
except Exception as e:
|
| 51 |
error_msg = f"Error: {str(e)}"
|
|
@@ -54,10 +54,10 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
|
|
| 54 |
|
| 55 |
def generate_coverage(progress=gr.Progress()):
|
| 56 |
console = ConsoleOutput()
|
| 57 |
-
logging.basicConfig(level=logging.DEBUG
|
| 58 |
logger = logging.getLogger()
|
| 59 |
logger.handlers = []
|
| 60 |
-
|
| 61 |
class GradioHandler(logging.Handler):
|
| 62 |
def emit(self, record):
|
| 63 |
msg = self.format(record)
|
|
@@ -78,7 +78,7 @@ def generate_coverage(progress=gr.Progress()):
|
|
| 78 |
coverage = f.read()
|
| 79 |
|
| 80 |
progress(1.0, desc="Complete!")
|
| 81 |
-
return coverage, console.get_output()
|
| 82 |
|
| 83 |
except Exception as e:
|
| 84 |
error_msg = f"Error: {str(e)}"
|
|
@@ -92,7 +92,7 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
|
|
| 92 |
file_input = gr.File(label="Upload Screenplay PDF", file_types=[".pdf"])
|
| 93 |
|
| 94 |
with gr.Row():
|
| 95 |
-
process_btn = gr.Button("Process Screenplay")
|
| 96 |
coverage_btn = gr.Button("Generate Coverage", interactive=False)
|
| 97 |
|
| 98 |
with gr.Row():
|
|
|
|
| 21 |
raise gr.Error("Please upload a PDF file")
|
| 22 |
|
| 23 |
console = ConsoleOutput()
|
| 24 |
+
logging.basicConfig(level=logging.DEBUG)
|
| 25 |
logger = logging.getLogger()
|
| 26 |
logger.handlers = []
|
| 27 |
+
|
| 28 |
class GradioHandler(logging.Handler):
|
| 29 |
def emit(self, record):
|
| 30 |
msg = self.format(record)
|
|
|
|
| 45 |
cleaned_text = f.read()
|
| 46 |
|
| 47 |
progress(1.0, desc="Complete!")
|
| 48 |
+
return [cleaned_text, gr.update(interactive=True, variant="primary"), console.get_output()]
|
| 49 |
|
| 50 |
except Exception as e:
|
| 51 |
error_msg = f"Error: {str(e)}"
|
|
|
|
| 54 |
|
| 55 |
def generate_coverage(progress=gr.Progress()):
|
| 56 |
console = ConsoleOutput()
|
| 57 |
+
logging.basicConfig(level=logging.DEBUG)
|
| 58 |
logger = logging.getLogger()
|
| 59 |
logger.handlers = []
|
| 60 |
+
|
| 61 |
class GradioHandler(logging.Handler):
|
| 62 |
def emit(self, record):
|
| 63 |
msg = self.format(record)
|
|
|
|
| 78 |
coverage = f.read()
|
| 79 |
|
| 80 |
progress(1.0, desc="Complete!")
|
| 81 |
+
return [coverage, console.get_output()]
|
| 82 |
|
| 83 |
except Exception as e:
|
| 84 |
error_msg = f"Error: {str(e)}"
|
|
|
|
| 92 |
file_input = gr.File(label="Upload Screenplay PDF", file_types=[".pdf"])
|
| 93 |
|
| 94 |
with gr.Row():
|
| 95 |
+
process_btn = gr.Button("Process Screenplay", variant="primary")
|
| 96 |
coverage_btn = gr.Button("Generate Coverage", interactive=False)
|
| 97 |
|
| 98 |
with gr.Row():
|