Update app.py

app.py

@@ -7,40 +7,33 @@ from langchain import PromptTemplate
 import os
 import tempfile
 
-def format_resume_to_yaml(api_key, file_info):
-    # Set the API key for OpenAI
-    os.environ['OPENAI_API_KEY'] = api_key
+# Updated imports for Gradio components
+from gradio.components import File, Textbox
 
-    # 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."
+def format_resume_to_yaml(api_key, file):
+    # Set the API key for OpenAI
+    os.environ['OPENAI_API_KEY'] = api_key
 
-    # 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}"
+    file_content = file.read()
 
     # Check if the file content is not empty
     if not file_content:
-        return "The uploaded file is empty."
+        raise ValueError("The uploaded file is empty.")
 
-    # 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
+    # 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
 
+    # Now we can use PyPDFLoader with the path to the temporary file
+    try:
         loader = PyPDFLoader(temp_file_path)
         docs = loader.load_and_split()  # This will return a list of text chunks from the PDF
-    except Exception as e:  # Adjusted for a broader range of exceptions
-        return f"An error occurred while processing the PDF: {e}"
+    except (IOError, PyPDF2.errors.PdfReaderError) as e:  # Handle potential PDF reading errors
+        raise ValueError(f"An error occurred while processing the PDF: {e}")
 
     # Combine the text chunks into a single string
     resume_text = " ".join(docs)
@@ -98,14 +91,18 @@ def format_resume_to_yaml(api_key, file_info):
     res = llm_chain.predict(human_input=resume_text)
     return res['output_text']
 
+def on_file_upload(filename, file_content):
+    if not file_content:
+        gr.Interface.alert(title="Error", message="Please upload a valid PDF resume.")
+
 def main():
-    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)
+    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")
 
     iface = gr.Interface(
         fn=format_resume_to_yaml,
-        inputs=[input_api_key, input_pdf_file],
+        inputs=[input_api_key, File(label="Upload your PDF resume", upload_event=on_file_upload)],
         outputs=output_yaml,
         title="Resume to YAML Formatter",
         description="Upload a PDF resume and enter your OpenAI API key to get it formatted to a YAML template.",
@@ -113,5 +110,6 @@ def main():
 
     iface.launch(debug=True)
 
+
 if __name__ == "__main__":
-    main()
+    main()