Update app.py

app.py

@@ -5,6 +5,27 @@ from PIL import Image
 from transformers import AutoTokenizer, AutoModel
 import torch
 
+
+
+# Define custom CSS styles
+button_style = gr.style(
+    padding="10px 20px",
+    background="#007BFF",
+    color="white",
+    border="none",
+    cursor="pointer",
+    font_size="16px",
+    margin="10px"
+)
+
+# Define layout with custom styles
+layout = [
+    gr.Row([gr.File(label="Upload PDF", type="file")]),
+    gr.Row([gr.Button("Generate Insights", type="submit", style=button_style)]),
+    gr.Row([gr.Textbox("Placeholder for PDF insights", label="Insights", type="text")])
+]
+
+
 # Function to get image embeddings using ViT
 def get_image_embeddings(image_path, model_name='google/vit-base-patch16-224'):
     feature_extractor = ViTFeatureExtractor.from_pretrained(model_name)
@@ -64,12 +85,11 @@ def process_pdf_and_generate_response(pdf_file):
     response = "Response based on the processed PDF"
     return response
 
-# Gradio interface
+# Gradio interface with customized layout
 iface = gr.Interface(
     fn=process_pdf_and_generate_response,
-    inputs=gr.inputs.File(label="Upload PDF", type="file"),
-    outputs=gr.outputs.Textbox(),
-    title="Talk2Deck - Interact with your PDFs",
+    inputs=layout,
+    title="pdf-chatbot",
     description="Upload a PDF and receive insights based on its content."
 )