Update app.py

app.py

@@ -6,38 +6,80 @@ from langchain.memory import ConversationBufferMemory
 from langchain import PromptTemplate
 import os
 import tempfile
-from gradio.components import File, Textbox
 
-def format_resume_to_yaml(api_key, file):
+def format_resume_to_yaml(api_key, file_info):
     # Set the API key for OpenAI
     os.environ['OPENAI_API_KEY'] = api_key
-    
-    if file is None:
+
+    # Extract the file path
+    file_path = file_info['path'] if isinstance(file_info, dict) and 'path' in file_info else None
+
+    if not file_path:
         return "Please upload a valid PDF resume."
 
-    file_content = file.read()
+    # Read the file content
+    try:
+        with open(file_path, 'rb') as file:
+            file_content = file.read()
+    except Exception as e:
+        return f"An error occurred while reading the file: {e}"
 
     # Check if the file content is not empty
     if not file_content:
         return "The uploaded file is empty."
 
-    # Save the uploaded file content to a temporary file
-    with tempfile.NamedTemporaryFile(delete=False, mode='wb+') as tmp_file:
-        tmp_file.write(file_content)
-        tmp_file.flush()
-        os.fsync(tmp_file.fileno())  # Ensure data is written to disk
-        temp_file_path = tmp_file.name
-
+    # Proceed with processing the file content
     try:
+        # Save the uploaded file content to a temporary file
+        with tempfile.NamedTemporaryFile(delete=False, mode='wb+') as tmp_file:
+            tmp_file.write(file_content)
+            tmp_file.flush()
+            os.fsync(tmp_file.fileno())  # Ensure data is written to disk
+            temp_file_path = tmp_file.name
+
         loader = PyPDFLoader(temp_file_path)
         docs = loader.load_and_split()  # This will return a list of text chunks from the PDF
-    except (IOError, Exception) as e:  # Adjusted for a broader range of exceptions
-        raise ValueError(f"An error occurred while processing the PDF: {e}")
+    except Exception as e:  # Adjusted for a broader range of exceptions
+        return f"An error occurred while processing the PDF: {e}"
 
     # Combine the text chunks into a single string
     resume_text = " ".join(docs)
 
-    template = """[Your YAML template here]"""
+    template = """Format the provided resume to this YAML template:
+    ---
+    name: ''
+    phoneNumbers:
+    - ''
+    websites:
+    - ''
+    emails:
+    - ''
+    dateOfBirth: ''
+    addresses:
+    - street: ''
+      city: ''
+      state: ''
+      zip: ''
+      country: ''
+    summary: ''
+    education:
+    - school: ''
+      degree: ''
+      fieldOfStudy: ''
+      startDate: ''
+      endDate: ''
+    workExperience:
+    - company: ''
+      position: ''
+      startDate: ''
+      endDate: ''
+    skills:
+    - name: ''
+    certifications:
+    - name: ''
+
+    {chat_history}
+    {human_input}"""
 
     prompt = PromptTemplate(
         input_variables=["chat_history", "human_input"],
@@ -57,9 +99,9 @@ def format_resume_to_yaml(api_key, file):
     return res['output_text']
 
 def main():
-    input_api_key = Textbox(label="Enter your OpenAI API Key")
-    input_pdf_file = File(label="Upload your PDF resume")
-    output_yaml = Textbox(label="Formatted Resume in YAML", interactive=False)
+    input_api_key = gr.components.Textbox(label="Enter your OpenAI API Key")
+    input_pdf_file = gr.components.File(label="Upload your PDF resume", type="file")
+    output_yaml = gr.components.Textbox(label="Formatted Resume in YAML", interactive=False)
 
     iface = gr.Interface(
         fn=format_resume_to_yaml,