Update app.py

app.py

@@ -37,13 +37,14 @@ class ChatBot:
             json=data,
         )
         response_json = response.json()
-        return response_text
+        return response_json
 
     def get_history(self) -> list:
         return self.contents
 
 # Streamlit app
-st.title("Image Chatbot with Google's Gemini API")
+st.set_page_config(layout="wide")
+st.title("World Model")
 
 st.sidebar.title("Upload Image or Take a Picture")
 uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
@@ -54,7 +55,6 @@ if uploaded_file is not None:
     buffered = io.BytesIO()
     image.save(buffered, format="JPEG")
     image_base64 = base64.b64encode(buffered.getvalue()).decode()
-    
     api_key = os.environ["GEMINI_API_KEY"] # st.sidebar.text_input("Enter your API key", type="password")
 
     if api_key: