Improved app structure

app.py

@@ -1,49 +1,58 @@
-import yaml
-from smolagents import CodeAgent, HfApiModel
-from smolagents.tools import Tool
-from tools.resumescraper import ResumeScraperTool
-from huggingface_hub import InferenceClient
+import gradio as gr
+import io
+from PyPDF2 import PdfReader
+from app import create_agent
 
-class FinalAnswerTool(Tool):
-    name = "final_answer"
-    description = "Use this tool to provide your final roast"
-    inputs = {
-        "answer": {
-            "type": "string",
-            "description": "The final roast for the resume"
-        }
-    }
-    output_type = "string"
+def extract_text_from_pdf(file_obj) -> str:
+    reader = PdfReader(file_obj)
+    text = ""
+    for page in reader.pages:
+        page_text = page.extract_text()
+        if page_text:
+            text += page_text + "\n"
+    return text
 
-    def forward(self, answer: str) -> str:
-        return answer
-
-def create_agent():
-    final_answer = FinalAnswerTool()
-    resume_scraper = ResumeScraperTool()
+def process_resume(input_method, resume_text, pdf_file):
+    if input_method == "Text":
+        text = resume_text
+    else:
+        if pdf_file is None:
+            return "No PDF uploaded."
+        # Check if pdf_file is a string (i.e. a file path) or a file-like object
+        if isinstance(pdf_file, str):
+            with open(pdf_file, "rb") as f:
+                file_bytes = f.read()
+        else:
+            file_bytes = pdf_file.read()
+        file_obj = io.BytesIO(file_bytes)
+        text = extract_text_from_pdf(file_obj)
     
-    # Instantiate HfApiModel using Qwen/Qwen2.5-Coder-32B-Instruct for roasting.
-    model = HfApiModel(
-        max_tokens=2096,
-        temperature=0.5,
-        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-        custom_role_conversions=None,
-    )
+    if not text.strip():
+        return "No resume text found."
     
-    # Create a dedicated InferenceClient using your public endpoint
-    client = InferenceClient("https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud")
-    # Override the model's client with our dedicated client
-    model.client = client
+    agent = create_agent()
+    # Instruct the agent to roast the resume using the resume text.
+    response = agent.run(f"Roast this resume: {text}")
+    return response
+
+
+def toggle_inputs(method):
+    if method == "Text":
+        return gr.update(visible=True), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), gr.update(visible=True)
 
-    with open("prompts.yaml", 'r') as stream:
-        prompt_templates = yaml.safe_load(stream)
-        
-    agent = CodeAgent(
-        model=model,
-        tools=[resume_scraper, final_answer],
-        max_steps=6,
-        verbosity_level=1,
-        prompt_templates=prompt_templates
-    )
+with gr.Blocks() as demo:
+    gr.Markdown("# Resume Roaster")
+    gr.Markdown("Enter your resume as text or upload a PDF to receive a humorous, professional roast!")
+    
+    input_method = gr.Radio(choices=["Text", "PDF"], label="Input Method", value="Text")
+    resume_text = gr.Textbox(label="Resume Text", lines=10, visible=True)
+    pdf_file = gr.File(label="Upload Resume PDF", file_types=[".pdf"], visible=False)
+    output = gr.Textbox(label="Roast Result", lines=10)
+    submit_btn = gr.Button("Roast It!")
+    
+    input_method.change(fn=toggle_inputs, inputs=input_method, outputs=[resume_text, pdf_file])
+    submit_btn.click(fn=process_resume, inputs=[input_method, resume_text, pdf_file], outputs=output)
     
-    return agent
+demo.launch(share=True)

prompts.yaml

@@ -1,16 +1,14 @@
 system_prompt: |
-  You are a witty professional roaster who roasts resumes.
-  Your job is to create a humorous roast of a resume based solely on its content.
-  Keep the tone light, semi-professional, and extremely entertaining, with puns and jokes.
-  Add Meta jokes for Resume being too big, too short, other improvements and roasting if there is something funny on there, like casual buzzwords.
-
+  You are a witty SEMI-professional roaster who analyzes resumes.
+  Your job is to create a humorous, professional roast of a resume.
+  Focus on teasing overly verbose descriptions, excessive buzzwords, and generic statements,
+  while keeping the tone light and appropriate for a professional setting.
 task_prompt: |
-  Given the provided resume details, craft a roast that:
-  1. References key sections such as Education, Projects, and Skills.
-  2. Is humorous yet semi-professional.
-  3. Outputs plain text without any code formatting.
-
+  Using the provided resume details, craft a roast that:
+  1. References key sections such as Summary, Experience, Education, and Skills.
+  2. Is humorous but not mean-spirited.
+  3. Maintains a professional tone.
+  4. Adds creative flair that makes the roast both entertaining and insightful.
 final_answer:
-  pre_messages: "Final Roast: "
-  post_messages: ""
-  instructions: "Output your final roast as plain text with no code blocks."
+  pre_messages: "Final Roast:"
+  post_messages: ""