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():
|