Update app.py

app.py

@@ -9,17 +9,26 @@ import google.generativeai as genai
 GOOGLE_API_KEY = "AIzaSyB6WzzQDu0TOPieyAm6nygPAfa6LBudqNo"
 genai.configure(api_key=GOOGLE_API_KEY)
 
+# Default prompt to ensure concise responses
+DEFAULT_PROMPT = (
+    "You are an AI assistant. Provide all responses in concise, easy-to-read points, "
+    "limiting the entire response to 5 lines."
+)
 
 def get_gemini_response(input_text, image_text=""):
     model = genai.GenerativeModel('gemini-1.5-flash')
+    combined_prompt = f"{DEFAULT_PROMPT}\n\n"
+
     if input_text and image_text:
-        response = model.generate_content([input_text, image_text])
+        combined_prompt += f"Prompt: {input_text}\nContent from PDF/Image: {image_text}"
     elif input_text:
-        response = model.generate_content(input_text)
+        combined_prompt += f"Prompt: {input_text}"
     elif image_text:
-        response = model.generate_content(image_text)
+        combined_prompt += f"Content from PDF/Image: {image_text}"
     else:
-        response = None
+        return "No valid input provided."
+
+    response = model.generate_content(combined_prompt)
     return response.text if response else "No valid input provided."
 
 
@@ -31,10 +40,10 @@ def extract_text_from_pdf(pdf_file):
     return text
 
 
-# Initialize our Streamlit app
+# Initialize Streamlit app
 st.set_page_config(page_title="Gemini Image and PDF Demo")
 
-st.header("Gemini Application")
+st.header("Chatbot Developed by Innoscope.ai")
 
 input_text = st.text_input("Input Prompt: ", key="input")
 uploaded_file = st.file_uploader("Choose an image or PDF...", type=["jpg", "jpeg", "png", "pdf"])