sync

cicero_jobs/app.py

@@ -24,10 +24,12 @@ def _():
     import httpx
     import marimo as mo
     import mistune
+
     #
-    #from htmltodocxpatch import html4docx as htmlpatch
+    from html4docx import HtmlToDocx
+    from pydantic import DocumentUrl
 
-    #HtmlToDocx = htmlpatch.HtmlToDocx
+    # HtmlToDocx = htmlpatch.HtmlToDocx
     from mistune.plugins.abbr import abbr
     from mistune.plugins.def_list import def_list
     from mistune.plugins.footnotes import footnotes
@@ -39,7 +41,9 @@ def _():
     return (
         Agent,
         BaseModel,
+        DocumentUrl,
         Field,
+        HtmlToDocx,
         abbr,
         def_list,
         footnotes,
@@ -82,7 +86,7 @@ def _(mo):
     return
 
 
-@app.function
+@app.function(hide_code=True)
 def create_job_application_prompt(resume: str, cover_letter: str, fi: str | None = None) -> str:
     return f"""
 <JobApplicationGenerationRequest>
@@ -256,70 +260,45 @@ def create_job_application_prompt(resume: str, cover_letter: str, fi: str | None
 def _(BaseModel, Field):
     class ApplicationMaterials(BaseModel):
         """Structured output for resume and cover letter generation."""
-        agent_has_spoken: bool = Field(..., description="Markdown Richly Formatted resume highlighting relevant experience for the job. 2 pages MAX, aim for 1.")
-        resume: str = Field(..., description="Markdown Richly Formatted resume highlighting relevant experience for the job. 2 pages MAX, aim for 1.")
-        cover_letter: str = Field(..., description="Markdown Richly Formatted and Compelling, personalized cover letter (3-4 paragraphs)")
-        letter_to_recruiter: str = Field(
-            ...,
-            description="Markdown Richly Formatted and Concise professional message for LinkedIn/email outreach (5-7 sentences)",
-        )
+
+        agent_has_spoken: bool = Field(..., description='Markdown Richly Formatted resume highlighting relevant experience for the job. 2 pages MAX, aim for 1.')
+        resume: str = Field(..., description='Markdown Richly Formatted resume highlighting relevant experience for the job. 2 pages MAX, aim for 1.')
+        cover_letter: str = Field(..., description='Markdown Richly Formatted and Compelling, personalized cover letter (3-4 paragraphs)')
+        letter_to_recruiter: str = Field(..., description='Markdown Richly Formatted and Concise professional message for LinkedIn/email outreach (5-7 sentences)')
         tips: str = Field(
             ...,
-            description="Markdown Richly Formatted and 5-7 actionable tips specific to this application, formatted as bullet points. MANDATORY: Must end wishing the candidate good luck and explaining finding a job is a numbers game. Motivate them!"
-
+            description='Markdown Richly Formatted and 5-7 actionable tips specific to this application, formatted as bullet points. MANDATORY: Must end wishing the candidate good luck and explaining finding a job is a numbers game. Motivate them!',
         )
     return (ApplicationMaterials,)
 
 
 @app.cell
 def _(mo):
-    resume_button = mo.ui.file(
-        kind="area",
-        filetypes=[".pdf", ".txt", ".doc", ".docx"],
-        label="Or drop your pdf/doc/docx/txt file résumé here!",
-    )
+    resume_button = mo.ui.file(kind='area', filetypes=['.pdf', '.txt', '.doc', '.docx'], label='Or drop your pdf/doc/docx/txt file résumé here!')
 
-    job_posting_area = mo.ui.text_area(placeholder="📋 Drop your job posting here!", full_width=True, rows=27)
+    job_posting_area = mo.ui.text_area(placeholder='📋 Drop your job posting here!', full_width=True, rows=27)
 
-    resume_area = mo.ui.text_area(
-        placeholder="📄 Provide the full content of your résumé. Uploading file in pdf/docx/doc/txt is preferred.",
-        full_width=True,
-        rows=20,
-    )
+    resume_area = mo.ui.text_area(placeholder='📄 Provide the full content of your résumé. Uploading file in pdf/docx/doc/txt is preferred.', full_width=True, rows=20)
 
-    job_url = mo.ui.text(
-        placeholder="https://example.com/job-posting",
-        label="Alternatively, provide the URL of the job posting",
-        full_width=True,
-    )
+    job_url = mo.ui.text(placeholder='https://example.com/job-posting', label='Alternatively, provide the URL of the job posting', full_width=True)
 
-    continue_button = mo.ui.run_button(
-        kind="success",
-        tooltip="Click submit to continue.",
-        label="Submit All (it takes a little bit, please be patient!)",
-        keyboard_shortcut="Ctrl+Enter",
-        full_width=True,
-    )
+    continue_button = mo.ui.run_button(kind='success', tooltip='Click submit to continue.', label='Submit All (it takes a little bit, please be patient!)', keyboard_shortcut='Ctrl+Enter', full_width=True)
 
     continue_render = mo.vstack(
         [
             mo.md("# Let's start with your materials!"),
             mo.vstack(
                 [
-                    mo.md("## "),
+                    mo.md('## '),
                     mo.hstack(
-                        [
-                            mo.vstack([mo.md("### Resume:"), resume_button, resume_area], align="center"),
-                            mo.vstack([mo.md("### Job Posting:"), job_posting_area, job_url], align="center"),
-                        ],
-                        align="center",
+                        [mo.vstack([mo.md('### Resume:'), resume_button, resume_area], align='center'), mo.vstack([mo.md('### Job Posting:'), job_posting_area, job_url], align='center')], align='center'
                     ),
                     continue_button,
                 ],
-                align="start",
+                align='start',
             ),
         ],
-        align="center",
+        align='center',
     )
     return (
         continue_button,
@@ -349,11 +328,11 @@ def _(
     resume_button,
 ):
     mo.stop(continue_button.value is not True)
-    if job_posting_area.value != "":
+    if job_posting_area.value != '':
         buceta = 1
         job_posting = str(job_posting_area.value)
         print(job_posting)
-    elif job_posting_area.value == "" and job_url.value == "":
+    elif job_posting_area.value == '' and job_url.value == '':
         buceta = None
         job_posting = None
         print(job_posting_area.value)
@@ -361,20 +340,24 @@ def _(
         buceta = None
         job_posting = None
         print(job_posting_area.value)
-    elif job_posting_area.value == "" and job_url.value != "":
+    elif job_posting_area.value is '' and job_url.value is not '':
         buceta = 1
+
         url_validation = DocumentUrl(job_url.value)
         print(url_validation)
-        url = f"https://r.jina.ai/{job_url.value}"
+        url = f'https://r.jina.ai/{job_url.value}'
 
-        jina_api_key = os.environ.get("JINA_API_KEY", "")
-        headers = {"Authorization": f"Bearer {jina_api_key}"}
+        jina_api_key = os.environ.get('JINA_API_KEY', '')
+        headers = {'Authorization': f'Bearer {jina_api_key}'}
         response = httpx.get(url, headers=headers)
         job_posting = response.text
         print(job_posting)
+    else:
+        job_posting = ''
+        buceta = None
 
     try:
-        if resume_button.contents() is not "" and resume_button.contents() is not None:
+        if resume_button.contents() is not '' and resume_button.contents() is not None:
             resume_contents = resume_button.contents()
             try:
                 from markitdown import MarkItDown as md_instance
@@ -384,7 +367,7 @@ def _(
                 docx_converted = md.convert(source=forcing_bytes)
                 resume_complete = str(docx_converted.text_content)
                 print(resume_complete)
-                print("0")
+                print('0')
             except Exception:
                 if isinstance(resume_contents, (bytes, bytearray)):
                     from markitdown import MarkItDown as md_instance
@@ -393,16 +376,16 @@ def _(
                     forcing_bytes = io.BytesIO(resume_contents)
                     docx_converted = md.convert(forcing_bytes)
                     resume_complete = str(docx_converted.text_content)
-                    print("1")
+                    print('1')
                 else:
                     resume_complete = resume_contents
-                    print("2")
+                    print('2')
         else:
-            resume_complete = ""
-            print("3")
+            resume_complete = ''
+            print('3')
     except Exception:
-        resume_complete = ""
-        print("4")
+        resume_complete = ''
+        print('4')
     return buceta, job_posting, resume_complete
 
 
@@ -419,39 +402,38 @@ async def _(
     mo.stop(continue_button.value is not True)
     mo.stop(buceta != 1)
     print(resume_complete, job_posting)
-    final_prompt = create_job_application_prompt(resume_complete, job_posting, fi="uga")
+    final_prompt = create_job_application_prompt(resume_complete, job_posting, fi='uga')
 
-    career_agent = Agent("gemini-2.5-flash", output_type=ApplicationMaterials)
+    career_agent = Agent('gemini-2.5-flash', output_type=ApplicationMaterials)
     result = await career_agent.run(final_prompt)
 
 
     import json
-    preview_section = mo.vstack(
-            [
-                mo.md("## Preview Your Documents"),
-                mo.md("---"),
-                mo.md("### Here is your résumé, how it looks like?"),
-                mo.md("---"),
-                mo.md(result.output.resume),
-                mo.md("---"),
-                mo.md("---"),
-                mo.md("### 💼 Here is your Cover Letter"),
-                mo.md("---"),
-                mo.md(result.output.cover_letter),
-                mo.md("---"),
-                mo.md("---"),
-                mo.md("### 📧 Here is your message to reach out to the recruiter"),
-                mo.md("---"),
-                mo.md(result.output.letter_to_recruiter),
-                mo.md("---"),
-                mo.md("---"),
-                mo.md("### 💡 Here are some additional tips"),
-                mo.md("---"),
-                mo.md(result.output.tips),
-                mo.md("---"),
-                mo.md("---"),
-            ]
-        )
+
+    preview_section = mo.vstack([
+        mo.md('## Preview Your Documents'),
+        mo.md('---'),
+        mo.md('### Here is your résumé, how it looks like?'),
+        mo.md('---'),
+        mo.md(result.output.resume),
+        mo.md('---'),
+        mo.md('---'),
+        mo.md('### 💼 Here is your Cover Letter'),
+        mo.md('---'),
+        mo.md(result.output.cover_letter),
+        mo.md('---'),
+        mo.md('---'),
+        mo.md('### 📧 Here is your message to reach out to the recruiter'),
+        mo.md('---'),
+        mo.md(result.output.letter_to_recruiter),
+        mo.md('---'),
+        mo.md('---'),
+        mo.md('### 💡 Here are some additional tips'),
+        mo.md('---'),
+        mo.md(result.output.tips),
+        mo.md('---'),
+        mo.md('---'),
+    ])
     agent_has_spoken = result.output.agent_has_spoken
     return agent_has_spoken, preview_section, result
 
@@ -489,71 +471,73 @@ def _(
     mo.stop(agent_has_spoken is not True)
     mo.stop(preview_section is None)
 
+
     def markdown_to_docx_bytes(markdown_text: str) -> bytes | None:
         """Synchronously converts markdown text to DOCX bytes with legal formatting."""
         try:
-            MISTUNE_PLUGINS = [
-                strikethrough,
-                footnotes,
-                table,
-                task_lists,
-                insert,
-                def_list,
-                abbr,
-                mark,
-                subscript,
-                superscript,
-            ]
-            # html_renderer = mistune.HTMLRenderer()
+            MISTUNE_PLUGINS = [strikethrough, footnotes, table, task_lists, insert, def_list, abbr, mark, subscript, superscript]
+            html_renderer = mistune.HTMLRenderer()
             # markdown_instance = mistune.html(plugins=MISTUNE_PLUGINS, renderer=html_renderer)
 
-            html_content = mistune.html(markdown_text, hard_wrap=True)
+            html_content = mistune.html(markdown_text)
             try:
                 parser = HtmlToDocx()
 
                 docx_document = parser.parse_html_string(html_content)
                 # Save to memory
 
-
                 out = io.BytesIO()
                 docx_document.save(out)
                 doc_bytes = out.getvalue()
-                print("uga0")
+                print('uga0')
                 return doc_bytes
             except Exception:
-                print("uga1")
+                print('uga1')
                 pass
 
         except Exception:
-            print("uga2")
+            print('uga2')
             pass
 
-    if result.output.resume is not None and result.output.resume is not "":
-        final_resume = markdown_to_docx_bytes(result.output.resume)
-        final_cover_letter = markdown_to_docx_bytes(result.output.cover_letter)
-        final_letter_to_recruiter = markdown_to_docx_bytes(result.output.letter_to_recruiter)
-
-        resume_download = mo.download(data=io.BytesIO(final_resume), filename="resume.docx", label="Résumé", mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
 
-        cover_letter_download = mo.download(data=io.BytesIO(final_cover_letter), filename="cover_letter.docx", label="Cover Letter", mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
-
-        message_download = mo.download(data=io.BytesIO(final_letter_to_recruiter), filename="letter_to_recruiter.docx", label="Recruiter Message", mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
-
-        downloads_section = mo.vstack(
-            [
-                mo.md("## 🎉 Your Documents Are Ready!"),
-                mo.md("Click the buttons below to download your documents:"),
-                mo.hstack([resume_download, cover_letter_download, message_download], justify="center", gap=2),
-                mo.md("---"),
-                mo.callout(mo.md("✅ **Success!** Your documents have been generated and are ready to download."), kind="success"),
-            ]
-        )
+    if result.output.resume is not None:
+        if result.output.resume is not '':
+            final_resume = markdown_to_docx_bytes(result.output.resume)
+            final_cover_letter = markdown_to_docx_bytes(result.output.cover_letter)
+            final_letter_to_recruiter = markdown_to_docx_bytes(result.output.letter_to_recruiter)
+            if final_resume is not None:
+                resume_download = mo.download(data=(final_resume), filename='resume.docx', label='Résumé', mimetype='application/vnd.openxmlformats-officedocument.wordprocessingml.document')
+            if final_cover_letter is not None:
+                cover_letter_download = mo.download(
+                    data=(final_cover_letter), filename='cover_letter.docx', label='Cover Letter', mimetype='application/vnd.openxmlformats-officedocument.wordprocessingml.document'
+                )
+            if final_letter_to_recruiter is not None:
+                message_download = mo.download(
+                    data=(final_letter_to_recruiter), filename='letter_to_recruiter.docx', label='Recruiter Message', mimetype='application/vnd.openxmlformats-officedocument.wordprocessingml.document'
+                )
+
+                downloads_section = mo.vstack([
+                    mo.md('## 🎉 Your Documents Are Ready!'),
+                    mo.md('Click the buttons below to download your documents:'),
+                    mo.hstack([resume_download, cover_letter_download, message_download], justify='center', gap=2),
+                    mo.md('---'),
+                    mo.callout(mo.md('✅ **Success!** Your documents have been generated and are ready to download.'), kind='success'),
+                ])
+                print('we have downloads-section')
+            else:
+                downloads_section = None
+                print('we dont have downloads-section')
+        else:
+            downloads_section = None
+            print('we dont have downloads-section')
     else:
         downloads_section = None
-    return downloads_section
-    
+        print('we dont have downloads-section')
+    return (downloads_section,)
+
+
 @app.cell
-def _(mo, buceta, agent_has_spoken, preview_section, downloads_section):
+def _(agent_has_spoken, buceta, downloads_section, mo, preview_section):
     mo.stop(buceta != 1)
     mo.stop(agent_has_spoken is not True)
     mo.stop(preview_section is None)
@@ -561,7 +545,9 @@ def _(mo, buceta, agent_has_spoken, preview_section, downloads_section):
     downloads_section
     return
 
-def _(mo, buceta, agent_has_spoken, preview_section, downloads_section):
+
+@app.cell
+def _(agent_has_spoken, buceta, downloads_section, mo, preview_section):
     mo.stop(buceta != 1)
     mo.stop(agent_has_spoken is not True)
     mo.stop(preview_section is None)
@@ -569,5 +555,6 @@ def _(mo, buceta, agent_has_spoken, preview_section, downloads_section):
     mo.md("""# <center>Time to do another application! Just add new materials. You got this!</center>""")
     return
 
+
 if __name__ == "__main__":
     app.run()

test_main.py

@@ -4,89 +4,90 @@ Unit tests for the main FastAPI application.
 
 import pytest
 from fastapi.testclient import TestClient
+
 from _server.main import app
 
 
 class TestRootEndpoint:
     """Test cases for the root endpoint."""
-    
+
     @pytest.fixture
     def client(self):
         """Create a test client for the FastAPI app."""
         return TestClient(app)
-    
+
     def test_root_redirects_to_cicero_jobs_app(self, client):
         """Test that the root endpoint "/" redirects to "/cicero_jobs/app"."""
-        response = client.get("/", follow_redirects=False)
-        
+        response = client.get('/', follow_redirects=False)
+
         # Assert it's a redirect response
         assert response.status_code == 307
-        
+
         # Assert the location header points to the correct URL
-        assert response.headers["location"] == "/cicero_jobs/app"
-    
+        assert response.headers['location'] == '/cicero_jobs/app'
+
     def test_root_redirect_response_type(self, client):
         """Test that the root endpoint returns a RedirectResponse."""
         response = client.get("/", follow_redirects=False)
-        
+
         # Check status code for temporary redirect
         assert response.status_code == 307
-        
+
         # Verify it's a proper redirect response
         assert "location" in response.headers
-    
+
     def test_root_redirect_follows_correctly(self, client):
         """Test that following the redirect leads to the expected path."""
         # First, get the redirect response
-        redirect_response = client.get("/", follow_redirects=False)
-        redirect_url = redirect_response.headers["location"]
-        
+        redirect_response = client.get('/', follow_redirects=False)
+        redirect_url = redirect_response.headers['location']
+
         # Then try to access the redirected URL
-        # Note: This might fail if the marimo app isn't fully set up, 
+        # Note: This might fail if the marimo app isn't fully set up,
         # but we're testing the redirect behavior
         final_response = client.get(redirect_url, follow_redirects=False)
-        
+
         # The redirect URL should be accessible (even if it returns an error due to marimo setup)
         # We're mainly testing that the redirect target exists in our app
-        assert redirect_url == "/cicero_jobs/app"
-        
+        assert redirect_url == '/cicero_jobs/app'
+
     def test_root_redirect_preserves_method(self, client):
         """Test that GET method is preserved in redirect."""
         response = client.get("/", follow_redirects=False)
-        
+
         assert response.status_code == 307  # Temporary redirect preserves method
-        assert response.headers["location"] == "/cicero_jobs/app"
+        assert response.headers['location'] == '/cicero_jobs/app'
 
 
 # Additional test class for edge cases
 class TestRootEndpointEdgeCases:
     """Test edge cases for the root endpoint."""
-    
+
     @pytest.fixture
     def client(self):
         """Create a test client for the FastAPI app."""
         return TestClient(app)
-    
+
     def test_root_with_query_parameters(self, client):
         """Test root endpoint behavior with query parameters."""
-        response = client.get("/?param=value", follow_redirects=False)
-        
+        response = client.get('/?param=value', follow_redirects=False)
+
         # Should still redirect regardless of query parameters
         assert response.status_code == 307
         assert response.headers["location"] == "/cicero_jobs/app"
-    
+
     def test_root_post_method(self, client):
         """Test that POST to root also redirects (if supported)."""
-        response = client.post("/", follow_redirects=False)
-        
+        response = client.post('/', follow_redirects=False)
+
         # FastAPI should either redirect or return method not allowed
         # Since we only defined GET, this should return 405 Method Not Allowed
         assert response.status_code == 405
-    
+
     def test_root_with_trailing_slash_consistency(self, client):
         """Test that root endpoint handles trailing slashes consistently."""
-        response1 = client.get("/", follow_redirects=False)
-        
+        response1 = client.get('/', follow_redirects=False)
+
         # Both should behave the same way
         assert response1.status_code == 307
         assert response1.headers["location"] == "/cicero_jobs/app"