Update app.py

app.py

@@ -59,6 +59,7 @@ num_criteria = len(queries)
 
 author_query = config_manager.get_config_value("model", "author_query")
 journal_query = config_manager.get_config_value("model", "journal_query")
+title_query = config_manager.get_config_value("model", "title_query")
 
 
 # Helper function to check if the file extension is allowed
@@ -175,11 +176,12 @@ def process_pdf(uploaded_file, llm_model, n_criteria = num_criteria):
     qa_author_prompt_with_example = info_prompt.replace("{few_shot_examples}", formatted_author_example)
     qa_journal_prompt_with_example = info_prompt.replace("{few_shot_examples}", formatted_journal_example)
 
-    info_llm = OpenAI(model="gpt-4-1106-preview", temperature=model_temperature, max_tokens=output_token_size)
+    info_llm = OpenAI(model="gpt-4-1106-preview", temperature=model_temperature, max_tokens=100)
     pdf_info_query = PDFQueryEngine(documents, info_llm, embed_model, (info_prompt))
     info_query_engine = pdf_info_query.setup_query_engine()
     journal_result = info_query_engine.query(modified_journal_query).response
     author_result = info_query_engine.query(author_query).response
+    title_result = info_query_engine.query(title_query).response
     
     
     pdf_criteria_query = PDFQueryEngine(documents, info_llm, embed_model, (general_prompt))
@@ -203,7 +205,7 @@ def process_pdf(uploaded_file, llm_model, n_criteria = num_criteria):
     score_bar_html = generate_score_bar(total_score, n_criteria)
 
     # Return the score as a string and the reasoning as HTML
-    return str(round((total_score / n_criteria) * 100)) + "/100", score_bar_html, reasoning_html, author_result
+    return str(round((total_score / n_criteria) * 100)) + "/100", score_bar_html, reasoning_html, author_result, title_result
 
 
 with gr.Blocks(theme=gr.themes.Glass(
@@ -226,6 +228,13 @@ with gr.Blocks(theme=gr.themes.Glass(
     score_bar_output = gr.HTML()
     reasoning_output = gr.HTML()
 
+    # Heading for Author Information
+    gr.Markdown("## Title of the paper")
+
+    # Output for dynamically generated author information
+    title_info_output = gr.Markdown()
+
+        
     # Heading for Author Information
     gr.Markdown("## Author Information")
 
@@ -236,7 +245,7 @@ with gr.Blocks(theme=gr.themes.Glass(
     submit_button.click(
         fn=process_pdf,
         inputs=[file_upload, model_choice],
-        outputs=[score_output, score_bar_output, reasoning_output, author_info_output]
+        outputs=[score_output, score_bar_output, reasoning_output, author_info_output, title_info_output]
     )