Update app.py

app.py

@@ -1,91 +1,59 @@
-import os
 import gradio as gr
 import cohere
+import os
 from docx import Document
 from docx.shared import Pt
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 import pypandoc
+from datetime import datetime
 
-# Initialize Cohere client with your API key from environment variable
-cohere_api_key = os.getenv('COHERE_API_KEY')  # Fetch the API key from environment variables
+# Get the API key from environment variables
+cohere_api_key = os.getenv('COHERE_API_KEY')
 if not cohere_api_key:
-    raise ValueError("Cohere API key not found. Please set the 'COHERE_API_KEY' environment variable.")
+    raise ValueError("Please set the COHERE_API_KEY environment variable")
 
+# Initialize Cohere client with your API key
 co = cohere.Client(cohere_api_key)
 
 def generate_body(job_description, language):
-    """
-    Generate the body of a job application letter using Cohere's language model.
-    
-    :param job_description: The description of the job you are applying for.
-    :param language: The language in which the letter should be written (English or German).
-    :return: The generated body text for the job application letter.
-    """
-    # Set the language model based on user selection
-    model = 'command-xlarge-nightly'  # Default to command-xlarge-nightly model for both English and German
-
-    # Modify the prompt to exclude greetings
+    model = 'command-xlarge-nightly'
+
     prompt = f"Write a professional job application letter in {language} without a greeting. Only generate the body text based on this job description:\n{job_description}"
 
-    # Use Cohere's API to generate the body of the application letter
     response = co.generate(
         model=model,
         prompt=prompt,
-        max_tokens=250,  # Reduced to ensure the content fits on one page
+        max_tokens=250,
         temperature=0.7,
     )
     return response.generations[0].text.strip()
 
-def create_application_letter(name, address, email, phone, job_position, employer_name, greeting_option, employer_contact_name, employer_address, job_id, start_date, job_description, language, output_format):
-    """
-    Create a job application letter with the provided details and format it for download.
-    
-    :param name: Applicant's full name.
-    :param address: Applicant's address.
-    :param email: Applicant's email address.
-    :param phone: Applicant's phone number.
-    :param job_position: Position being applied for.
-    :param employer_name: Employer's name.
-    :param greeting_option: Indicates whether the recipient's name is known.
-    :param employer_contact_name: Employer contact's name (if known).
-    :param employer_address: Employer's address.
-    :param job_id: Job ID for the application.
-    :param start_date: Desired start date for the job.
-    :param job_description: The job description.
-    :param language: Language of the letter (English or German).
-    :param output_format: Desired output format (DOCX or PDF).
-    :return: The filename of the generated application letter.
-    """
-    # Generate the body using the job description and language
+def create_application_letter(name, sender_street, sender_zip, sender_city, email, phone, job_position, employer_name, employer_street, employer_zip, employer_city, greeting_option, employer_contact_name, job_id, start_date, job_description, language, output_format):
     body = generate_body(job_description, language)
 
-    # Create a new Document
     doc = Document()
 
-    # Add header with sender's name, address, email, and phone in a single line
     header = doc.sections[0].header
     header_paragraph = header.paragraphs[0]
-    header_paragraph.text = f"{name} | {address} | {email} | {phone}"
+    header_paragraph.text = f"{name} | {sender_street}, {sender_zip} {sender_city} | {email} | {phone}"
     header_paragraph.alignment = WD_PARAGRAPH_ALIGNMENT.LEFT
 
-    # Adjust the font size for the header
     for run in header_paragraph.runs:
         run.font.size = Pt(10)
 
-    # Add one blank line above the employer's information
-    doc.add_paragraph("\n")
+    doc.add_paragraph(f"{employer_name}\n{employer_contact_name if greeting_option == 'Known' else ''}\n{employer_street}\n{employer_zip} {employer_city}")
 
-    # Add the date and employer's information in the main document
-    doc.add_paragraph(f"{employer_name}\n{employer_contact_name if greeting_option == 'Known' else ''}\n{employer_address}")
-    doc.add_paragraph(f"{address.split(',')[1]}, {start_date}\n\n")  # Assuming the city is part of the address
+    current_date = datetime.now().strftime('%d.%m.%Y')
+    paragraph = doc.add_paragraph()
+    paragraph.alignment = WD_PARAGRAPH_ALIGNMENT.RIGHT
+    run = paragraph.add_run(f"{employer_city}, {current_date}")
+    run.font.size = Pt(11)
 
-    # Add the subject
     doc.add_paragraph(f"Bewerbung als {job_position}\nKennnummer {job_id}\n", style='Heading 2')
 
-    # Add the greeting based on the selected option
     if language == "German":
         if greeting_option == "Known" and employer_contact_name:
-            doc.add_paragraph(f"Sehr geehrter Herr {employer_contact_name.split()[0]},\n")
+            doc.add_paragraph(f"Sehr geehrter Herr {employer_contact_name},\n")
         else:
             doc.add_paragraph("Sehr geehrte Damen und Herren,\n")
     else:
@@ -94,10 +62,8 @@ def create_application_letter(name, address, email, phone, job_position, employe
         else:
             doc.add_paragraph("Dear Sir/Madam,\n")
 
-    # Add the generated body of the letter
     doc.add_paragraph(body)
 
-    # Add closing
     closing_text = (
         f"\nIch unterstütze Ihr Team gerne ab dem {start_date} und freue mich über die Einladung zu einem persönlichen Vorstellungsgespräch."
         if language == "German" else
@@ -107,63 +73,59 @@ def create_application_letter(name, address, email, phone, job_position, employe
     doc.add_paragraph("\nMit freundlichen Grüßen,\n\n" if language == "German" else "\nSincerely,\n\n")
     doc.add_paragraph(f"{name}\n")
 
-    # Adjust font size for body to ensure it fits on one page
     for paragraph in doc.paragraphs:
         for run in paragraph.runs:
-            run.font.size = Pt(11)  # Slightly reduced to ensure one-page fit
+            run.font.size = Pt(11)
 
-    # Save the document
     output_filename_docx = f'{name}_application_letter.docx'
     doc.save(output_filename_docx)
 
-    # Convert to PDF if requested
     if output_format == "PDF":
         output_filename_pdf = f'{name}_application_letter.pdf'
         pypandoc.convert_file(output_filename_docx, 'pdf', outputfile=output_filename_pdf)
-        os.remove(output_filename_docx)  # Optionally remove the DOCX file after conversion
+        os.remove(output_filename_docx)
         return output_filename_pdf
     else:
         return output_filename_docx
 
-def generate_and_download(name, address, email, phone, job_position, employer_name, greeting_option, employer_contact_name, employer_address, job_id, start_date, job_description, language, output_format):
-    """
-    Generate the application letter and return the file for download.
-    
-    :return: Filename of the generated application letter for download.
-    """
-    output_filename = create_application_letter(name, address, email, phone, job_position, employer_name, greeting_option, employer_contact_name, employer_address, job_id, start_date, job_description, language, output_format)
+def generate_and_download(name, sender_street, sender_zip, sender_city, email, phone, job_position, employer_name, employer_street, employer_zip, employer_city, greeting_option, employer_contact_name, job_id, start_date, job_description, language, output_format):
+    output_filename = create_application_letter(name, sender_street, sender_zip, sender_city, email, phone, job_position, employer_name, employer_street, employer_zip, employer_city, greeting_option, employer_contact_name, job_id, start_date, job_description, language, output_format)
     return output_filename
 
-# Define the Gradio interface
+description = """
+# Professional Job Application Letter Generator
+
+Welcome to the ultimate tool for crafting professional job application letters with ease. This tool leverages the power of **Cohere**, a state-of-the-art language model, to automatically generate a tailored application letter body based on the job description you provide. 
+
+Simply fill in the required details, and within seconds, you'll have a beautifully formatted job application letter ready to download in either DOCX or PDF format.
+
+Start your journey to your next career opportunity with a compelling application letter generated just for you!
+"""
+
 with gr.Blocks() as demo:
-    gr.Markdown("""
-    ## Professional Job Application Letter Generator
-    
-    This tool helps you generate a professional job application letter in either English or German. 
-    The letter is automatically formatted and can be downloaded in DOCX or PDF format.
-    
-    ### Powered by Cohere's Language Model
-    
-    The body of the application letter is generated using **Cohere**, a state-of-the-art language model, 
-    ensuring that your application is not only well-written but also tailored to the job description you provide.
-    """)
+    gr.Markdown(description)
     
-    name = gr.Textbox(label="Name", placeholder="Enter your full name", value="Claire Waßer")
-    address = gr.Textbox(label="Address", placeholder="Enter your address", value="Musterstraße 78, 23456 Musterstadt")
-    email = gr.Textbox(label="Email", placeholder="Enter your email", value="[email protected]")
-    phone = gr.Textbox(label="Phone", placeholder="Enter your phone number", value="0171 23456789")
-    job_position = gr.Textbox(label="Job Position", placeholder="Enter the job position", value="Physiotherapeutin")
-    employer_name = gr.Textbox(label="Employer Name", placeholder="Enter the employer's name", value="Arbeitgeber GmbH")
-    greeting_option = gr.Dropdown(choices=["Known", "Unknown"], label="Is the recipient's name known?", value="Unknown")
-    employer_contact_name = gr.Textbox(label="Employer Contact Name", placeholder="Enter the contact person's name (if known)", value="Name Nachname", visible=False)
-    employer_address = gr.Textbox(label="Employer Address", placeholder="Enter the employer's address", value="Straße 123, 12345 Musterstadt")
-    job_id = gr.Textbox(label="Job ID", placeholder="Enter the job ID", value="123456")
-    start_date = gr.Textbox(label="Start Date", placeholder="Enter the start date (e.g., 15.05.2020)", value="18.08.2024")
-    job_description = gr.Textbox(label="Job Description", placeholder="Paste the job description here", lines=7, value="Aktuell stehe ich am Ende meiner Berufsausbildung zur staatlich anerkannten Physiotherapeutin...")
+    name = gr.Textbox(label="Name", placeholder="Enter your full name", value="Maximilian Müller")
+    sender_street = gr.Textbox(label="Sender Street", placeholder="Enter your street address", value="Beispielstraße 42")
+    sender_zip = gr.Textbox(label="Sender Postal Code", placeholder="Enter your postal code", value="10115")
+    sender_city = gr.Textbox(label="Sender City", placeholder="Enter your city", value="Berlin")
+    email = gr.Textbox(label="Email", placeholder="Enter your email", value="[email protected]")
+    phone = gr.Textbox(label="Phone", placeholder="Enter your phone number", value="+49 170 1234567")
+
+    job_position = gr.Textbox(label="Job Position", placeholder="Enter the job position", value="Softwareentwickler")
+    employer_name = gr.Textbox(label="Employer Name", placeholder="Enter the employer's name", value="Tech Innovations GmbH")
+    employer_street = gr.Textbox(label="Employer Street", placeholder="Enter the employer's street address", value="Innovationsstraße 1")
+    employer_zip = gr.Textbox(label="Employer Postal Code", placeholder="Enter the employer's postal code", value="10115")
+    employer_city = gr.Textbox(label="Employer City", placeholder="Enter the employer's city", value="Berlin")
+    greeting_option = gr.Dropdown(choices=["Known", "Unknown"], label="Is the recipient's name known?", value="Known")
+    employer_contact_name = gr.Textbox(label="Employer Contact Name", placeholder="Enter the contact person's name (if known)", value="Herr Dr. Felix Schmidt", visible=True)
+
+    job_id = gr.Textbox(label="Job ID", placeholder="Enter the job ID", value="DEV-2024-01")
+    start_date = gr.Textbox(label="Start Date", placeholder="Enter the start date (e.g., 15.05.2020)", value="01.09.2024")
+    job_description = gr.Textbox(label="Job Description", placeholder="Paste the job description here", lines=7, value="We are seeking a dedicated Software Developer to join our team in creating innovative applications. Experience in Python and Java, as well as knowledge of agile methodologies, is required.")
     language = gr.Dropdown(choices=["English", "German"], label="Select Language", value="German")
     output_format = gr.Dropdown(choices=["DOCX", "PDF"], label="Select Output Format", value="PDF")
 
-    # Show or hide the employer contact name input based on the greeting option
     def update_visibility(greeting_option):
         return gr.update(visible=(greeting_option == "Known"))
 
@@ -173,8 +135,7 @@ with gr.Blocks() as demo:
     output = gr.File(label="Download your application letter")
 
     generate_button.click(generate_and_download,
-                          inputs=[name, address, email, phone, job_position, employer_name, greeting_option, employer_contact_name, employer_address, job_id, start_date, job_description, language, output_format],
+                          inputs=[name, sender_street, sender_zip, sender_city, email, phone, job_position, employer_name, employer_street, employer_zip, employer_city, greeting_option, employer_contact_name, job_id, start_date, job_description, language, output_format],
                           outputs=output)
 
-# Launch the Gradio interface
 demo.launch(debug=True)