Update app.py

app.py

@@ -21,7 +21,12 @@ def generate_summary(file, prompt, model, output_format, temperature, top_p):
         return "Please provide a prompt."
 
     file_text = file_to_string(file) if file else ""
-    full_prompt = f"{prompt}\n{file_text}" if file_text else prompt
+    
+    # Check if the {PDF_TEXT} token is in the prompt
+    if "{PDF_TEXT}" in prompt and file_text:
+        full_prompt = prompt.replace("{PDF_TEXT}", file_text)
+    else:
+        full_prompt = f"{prompt}\n{file_text}" if file_text else prompt
 
     try:
         client = Groq(api_key=os.environ["GROQ_API_KEY"])
@@ -62,7 +67,7 @@ with gr.Blocks() as iface:
     
     with gr.Row():
         file_input = gr.File(label="Upload File (Optional)")
-        prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...", lines=3)
+        prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here... Use {PDF_TEXT} to insert PDF content at a specific location.", lines=3)
     
     with gr.Row():
         model_dropdown = gr.Dropdown(label="Choose Model", choices=["llama-3.1-8b-instant", "llama-3.1-70b-versatile"], value="llama-3.1-70b-versatile")