reyemhorts commited on
Commit
b41f71a
·
1 Parent(s): 2c3cd08
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -4,8 +4,8 @@ from pypdf import PdfReader
4
  from typing import Optional
5
  import json
6
 
7
- from load_db import load_vectorestore_from_pdf
8
- from oai import chat_prompt, get_comparison
9
  from io import BytesIO
10
 
11
  TEMP_PDF_PATH = "temp.pdf"
@@ -13,17 +13,17 @@ retriever = None
13
  db = None
14
  documents = None
15
 
16
- with open("all_unique_modules.json","r",encoding="UTF-8") as f:
17
- courses = json.load(f)
18
 
19
- def get_course(code:str)->str|None:
20
- #print(len(courses))
21
- for c in courses:
22
- if not "Kurscode" in c:
23
- continue
24
- if c["Kurscode"].strip()==code.strip():
25
- return f"Kursbeschreibung:\n{c['Description']}\nKursziele:\n{c['Kursziele']}\nKursinhalte:\n{c['Kursinhalt']}"
26
- return None
27
 
28
  def pdf_to_text(file_path, page_num:Optional[int]=None):
29
  reader = PdfReader(file_path)
@@ -35,10 +35,10 @@ def pdf_to_text(file_path, page_num:Optional[int]=None):
35
  text += page_text
36
  return text
37
 
38
- def load_vectore_store():
39
- global retriever, db
40
- db = load_vectorestore_from_pdf(TEMP_PDF_PATH,persist=False)
41
- retriever = db.as_retriever(search_kwargs={"k": 4})
42
 
43
  def load_pdf(inp):
44
  # Convert bytes back to a PDF file
 
4
  from typing import Optional
5
  import json
6
 
7
+ #from load_db import load_vectorestore_from_pdf
8
+ #from oai import chat_prompt, get_comparison
9
  from io import BytesIO
10
 
11
  TEMP_PDF_PATH = "temp.pdf"
 
13
  db = None
14
  documents = None
15
 
16
+ # with open("all_unique_modules.json","r",encoding="UTF-8") as f:
17
+ # courses = json.load(f)
18
 
19
+ # def get_course(code:str)->str|None:
20
+ # #print(len(courses))
21
+ # for c in courses:
22
+ # if not "Kurscode" in c:
23
+ # continue
24
+ # if c["Kurscode"].strip()==code.strip():
25
+ # return f"Kursbeschreibung:\n{c['Description']}\nKursziele:\n{c['Kursziele']}\nKursinhalte:\n{c['Kursinhalt']}"
26
+ # return None
27
 
28
  def pdf_to_text(file_path, page_num:Optional[int]=None):
29
  reader = PdfReader(file_path)
 
35
  text += page_text
36
  return text
37
 
38
+ # def load_vectore_store():
39
+ # global retriever, db
40
+ # db = load_vectorestore_from_pdf(TEMP_PDF_PATH,persist=False)
41
+ # retriever = db.as_retriever(search_kwargs={"k": 4})
42
 
43
  def load_pdf(inp):
44
  # Convert bytes back to a PDF file