GurgenGulay commited on
Commit
06a6d14
·
verified ·
1 Parent(s): 6156752

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -23
app.py CHANGED
@@ -1,21 +1,12 @@
1
  import logging
2
  import gradio as gr
3
- from utils import process_large_text, generate_lesson_from_transcript as generate_lesson_from_transcript_logic
4
- from pdfminer.high_level import extract_text
5
 
6
 
7
- # Logging Ayarları
8
  logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
9
  logger = logging.getLogger(__name__)
10
 
11
- def pdf_to_text(pdf_path):
12
- try:
13
- logger.info(f"Extracting text from PDF: {pdf_path}")
14
- return extract_text(pdf_path)
15
- except Exception as e:
16
- logger.error(f"Error while extracting text from PDF: {str(e)}")
17
- raise ValueError(f"PDF extraction error: {str(e)}")
18
-
19
  def generate_lesson(doc_text=None, pdf_file=None):
20
  try:
21
  if pdf_file and doc_text:
@@ -26,18 +17,18 @@ def generate_lesson(doc_text=None, pdf_file=None):
26
  logger.info(f"Processing uploaded PDF file: {pdf_file.name}")
27
  doc_text = pdf_to_text(pdf_file.name)
28
 
29
- logger.info("Processing the document text with general model.")
30
- general_model_output = process_large_text(doc_text)
31
-
32
- logger.info("Refining the output with fine-tuned model.")
33
- refined_output = refine_with_fine_tuned_model(general_model_output)
34
-
35
- output_path = "/tmp/refined_output.txt"
36
- with open(output_path, "w") as file:
37
- file.write(refined_output)
38
-
39
- logger.info(f"Lesson generated successfully. Output saved to: {output_path}")
40
- return refined_output, gr.File(output_path)
41
 
42
  except Exception as e:
43
  logger.error(f"Error occurred while generating lesson: {str(e)}")
 
1
  import logging
2
  import gradio as gr
3
+ from utils import process_large_text, generate_lesson_from_transcript as generate_lesson_from_transcript_logic, pdf_to_text
4
+ from pdfminer.high_level import extract_text
5
 
6
 
 
7
  logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
8
  logger = logging.getLogger(__name__)
9
 
 
 
 
 
 
 
 
 
10
  def generate_lesson(doc_text=None, pdf_file=None):
11
  try:
12
  if pdf_file and doc_text:
 
17
  logger.info(f"Processing uploaded PDF file: {pdf_file.name}")
18
  doc_text = pdf_to_text(pdf_file.name)
19
 
20
+ logger.info("Processing the document text.")
21
+ processed_text = process_large_text(doc_text)
22
+
23
+ logger.info("Generating lesson from transcript.")
24
+ generated_text, output_path = generate_lesson_from_transcript_logic(processed_text)
25
+
26
+ if output_path:
27
+ logger.info(f"Lesson generated successfully. Output saved to: {output_path}")
28
+ return generated_text, gr.File(output_path)
29
+ else:
30
+ logger.info("Lesson generated successfully, but no output file created.")
31
+ return generated_text, None
32
 
33
  except Exception as e:
34
  logger.error(f"Error occurred while generating lesson: {str(e)}")