Update app.py

app.py

@@ -74,13 +74,13 @@ def txt_to_html(text):
     html_content += "</body></html>"
     return html_content
 
-def craft_cv(llm, cv_text, job_description, maxtokens, temperature, top_probability):
-# def craft_cv(llm, prompt, maxtokens, temperature, top_probability):
+# def craft_cv(llm, cv_text, job_description, maxtokens, temperature, top_probability):
+def craft_cv(llm, prompt, maxtokens, temperature, top_probability):
     instruction = "Given input CV and job description. Please revise the CV according to the given job description and output the revised CV."
     output = llm.create_chat_completion(
         messages=[
-            {"from": "user", "value": instruction + ' Input CV: ' + cv_text + ' , Job Description: ' + job_description},
-            # {"from": "user", "value": prompt},
+            # {"from": "user", "value": instruction + ' Input CV: ' + cv_text + ' , Job Description: ' + job_description},
+            {"from": "user", "value": prompt},
         ],
         max_tokens=maxtokens,
         temperature=temperature
@@ -153,9 +153,9 @@ def convert_to_json(llm, cv_text, maxtokens, temperature, top_probability):
     output = output['choices'][0]['message']['content']
     return output 
 
-# def pdf_to_text(prompt, maxtokens=2048, temperature=0, top_probability=0.95):
 @spaces.GPU(duration=40)
-def pdf_to_text(cv_file, job_description, maxtokens=2048, temperature=0, top_probability=0.95):
+def pdf_to_text(prompt, maxtokens=2048, temperature=0, top_probability=0.95):
+# def pdf_to_text(cv_file, job_description, maxtokens=2048, temperature=0, top_probability=0.95):
     
     # page2content = process_document(cv_file)
     # cv_text = ""
@@ -172,8 +172,8 @@ def pdf_to_text(cv_file, job_description, maxtokens=2048, temperature=0, top_pro
         n_batch=1024,
         n_ctx=8192,
     )
-    cv_text, crafted_cv = craft_cv(llm, cv_text, job_description, maxtokens, temperature, top_probability)
-    # cv_text, crafted_cv = craft_cv(llm, prompt, maxtokens, temperature, top_probability)
+    # cv_text, crafted_cv = craft_cv(llm, cv_text, job_description, maxtokens, temperature, top_probability)
+    cv_text, crafted_cv = craft_cv(llm, prompt, maxtokens, temperature, top_probability)
     crafted_cv = convert_to_json(llm, crafted_cv, maxtokens, temperature, top_probability)
     return crafted_cv
 
@@ -185,8 +185,8 @@ prompt_text = gr.Textbox(label='Enter the job description')
 output_text = gr.Textbox()
 iface = gr.Interface(
     fn=pdf_to_text,
-    inputs=[cv_file, prompt_text],
-    # inputs=['text'],
+    # inputs=[cv_file, prompt_text],
+    inputs=['text'],
     outputs=['text'],
     title='Craft CV',
     description="This application assists to customize CV based on input job description",