Update app.py

app.py

@@ -23,7 +23,6 @@ def format_prompt(message, history, system_prompt=None):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-# Generate text
 def generate(
     prompt, history, system_prompt=None, temperature=0.2, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0,
 ):
@@ -76,32 +75,29 @@ def save_file(content, filename, file_format):
     else:
         raise ValueError("Unsupported file format")
 
-# Combine generate and save file functions
 def generate_and_save(prompt, history, filename="output", file_format="pdf", system_prompt=None, temperature=0.2, max_new_tokens=1024, top_p=0.95, repetition_penalty=1.0):
     generated_text = ""
     for output in generate(prompt, history, system_prompt, temperature, max_new_tokens, top_p, repetition_penalty):
         generated_text = output
-    # Ensure </s> tags are removed from the final output
     generated_text = generated_text.replace("</s>", "")
     saved_file = save_file(generated_text, filename, file_format)
     return generated_text, history + [(prompt, generated_text)], saved_file
-
-# Create Gradio interface
+    
 demo = gr.Interface(
     fn=generate_and_save,
     inputs=[
-        gr.Textbox(placeholder="Type your message here...", label="Prompt"),
-        gr.State(value=[]),  # history
+        gr.Textbox(placeholder="Type your message here...", label="Chatbot", line=4),
+        gr.State(value=[]), 
         gr.Textbox(placeholder="Filename (default: output)", label="Filename", value="output"),
         gr.Radio(["pdf", "docx", "txt"], label="File Format", value="pdf"),
     ],
     outputs=[
-        gr.Textbox(label="Generated Text"),
-        gr.State(value=[]),  # history
+        gr.Textbox(label="Generated Text", line=4),
+        gr.State(value=[]), 
         gr.File(label="Download File")
     ],
     css=css,
-    title="",
+    title="GRAB DOC",
     theme="bethecloud/storj_theme"
 )