yalrashed commited on
Commit
11d6216
·
verified ·
1 Parent(s): 039925e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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, format='%(asctime)s - %(levelname)s - %(message)s')
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, format='%(asctime)s - %(levelname)s - %(message)s')
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():