Update app.py

app.py

@@ -295,7 +295,11 @@ class GradioUI:
             print(f"Error in interaction: {str(e)}")
             raise
 
-    def upload_file(self, file, file_uploads_log):
+    def upload_file(
+        self,
+        file,
+        file_uploads_log,
+    ):
         """Handle file uploads with proper validation and security."""
         if file is None:
             return gr.Textbox("No file uploaded", visible=True), file_uploads_log
@@ -310,7 +314,9 @@ class GradioUI:
 
         # Sanitize file name
         original_name = os.path.basename(file.name)
-        sanitized_name = re.sub(r"[^\w\-.]", "_", original_name)
+        sanitized_name = re.sub(
+            r"[^\w\-.]", "_", original_name
+        )  # Replace invalid chars with underscores
 
         # Ensure the extension correlates to the mime type
         type_to_ext = {}
@@ -325,11 +331,15 @@ class GradioUI:
 
         # Limit File Size, and Throw Error
         max_file_size_mb = 50  # Define the limit
-        file_size_mb = os.path.getsize(file.name) / (1024 * 1024)
+        file_size_mb = os.path.getsize(file.name) / (1024 * 1024)  # Size in MB
 
         if file_size_mb > max_file_size_mb:
-            error_msg = f"File size exceeds {max_file_size_mb} MB limit."
-            return gr.Textbox(error_msg, visible=True), file_uploads_log
+            return (
+                gr.Textbox(
+                    f"File size exceeds {max_file_size_mb} MB limit.", visible=True
+                ),
+                file_uploads_log,
+            )
 
         # Save the uploaded file to the specified folder
         file_path = os.path.join(self.file_upload_folder, sanitized_name)
@@ -344,17 +354,14 @@ class GradioUI:
         message = text_input
 
         if file_uploads_log:
-            message += (
-                "\nYou have been provided with these files, which might be helpful"
-                f" or not: {file_uploads_log}"
-            )
+            message += f"\nYou have been provided with these files, which might be helpful or not: {file_uploads_log}"  # Added file list
 
         return (
             message,
             gr.Textbox(
                 value="",
                 interactive=False,
-                placeholder="Processing...",
+                placeholder="Processing...",  # Changed placeholder.
             ),
             gr.Button(interactive=False),
         )
@@ -362,26 +369,29 @@ class GradioUI:
     def detect_device(self, request: gr.Request):
         """Detect whether the user is on mobile or desktop device."""
         if not request:
-            return "Unknown device"
+            return "Unknown device"  # Handle case where request is none.
 
+        # Method 1: Check sec-ch-ua-mobile header
         is_mobile_header = request.headers.get("sec-ch-ua-mobile")
         if is_mobile_header:
             return "Mobile" if "?1" in is_mobile_header else "Desktop"
 
+        # Method 2: Check user-agent string
         user_agent = request.headers.get("user-agent", "").lower()
         mobile_keywords = ["android", "iphone", "ipad", "mobile", "phone"]
 
         if any(keyword in user_agent for keyword in mobile_keywords):
             return "Mobile"
 
+        # Method 3: Check platform
         platform = request.headers.get("sec-ch-ua-platform", "").lower()
         if platform:
             if platform in ['"android"', '"ios"']:
                 return "Mobile"
-            # Remove unnecessary elif
-            if platform in ['"windows"', '"macos"', '"linux"']:
+            elif platform in ['"windows"', '"macos"', '"linux"']:
                 return "Desktop"
 
+        # Default case if no clear indicators
         return "Desktop"
 
     def launch(self, **kwargs):
@@ -395,8 +405,8 @@ class GradioUI:
                 # Render layout with sidebar
                 if device == "Desktop":
                     return self._create_desktop_layout()
-                # Replace unnecessary else
-                return self._create_mobile_layout()
+                else:
+                    return self._create_mobile_layout()
 
         demo.queue(max_size=20).launch(
             debug=True, **kwargs
@@ -421,7 +431,7 @@ class GradioUI:
                     launch_research_btn = gr.Button("Run", variant="primary")
 
                 # If an upload folder is provided, enable the upload feature
-                if self.file_upload_folder:
+                if self.file_upload_folder is not None:
                     upload_file = gr.File(label="Upload a file")
                     upload_status = gr.Textbox(
                         label="Upload Status", interactive=False, visible=False
@@ -497,7 +507,7 @@ class GradioUI:
             )
 
             # If an upload folder is provided, enable the upload feature
-            if self.file_upload_folder:
+            if self.file_upload_folder is not None:
                 upload_file = gr.File(label="Upload a file")
                 upload_status = gr.Textbox(
                     label="Upload Status", interactive=False, visible=False
@@ -536,49 +546,44 @@ class GradioUI:
         session_state,
     ):
         """Connect the event handlers for input elements."""
-
-        # Helper function to avoid duplicated code.
-        def process_message():
-            return self.log_user_message(text_input, file_uploads_log)
-
-        def interact():
-            return self.interact_with_agent(stored_messages, chatbot, session_state)
-
-        def reset_input():
-            return (
-                gr.Textbox(
-                    interactive=True,
-                    placeholder="Enter your prompt here and press the button",
-                ),
-                gr.Button(interactive=True),
-            )
-
         # Connect text input submit event
         text_input.submit(
-            process_message,
+            self.log_user_message,
             [text_input, file_uploads_log],
             [stored_messages, text_input, launch_research_btn],
         ).then(
-            interact,
+            self.interact_with_agent,
             [stored_messages, chatbot, session_state],
             [chatbot],
         ).then(
-            reset_input,
+            lambda: (
+                gr.Textbox(
+                    interactive=True,
+                    placeholder="Enter your prompt here and press the button",
+                ),
+                gr.Button(interactive=True),
+            ),
             None,
             [text_input, launch_research_btn],
         )
 
         # Connect button click event
         launch_research_btn.click(
-            process_message,
+            self.log_user_message,
             [text_input, file_uploads_log],
             [stored_messages, text_input, launch_research_btn],
         ).then(
-            interact,
+            self.interact_with_agent,
             [stored_messages, chatbot, session_state],
             [chatbot],
         ).then(
-            reset_input,
+            lambda: (
+                gr.Textbox(
+                    interactive=True,
+                    placeholder="Enter your prompt here and press the button",
+                ),
+                gr.Button(interactive=True),
+            ),
             None,
             [text_input, launch_research_btn],
         )