Update app.py

app.py

@@ -172,11 +172,13 @@ def respond(message, history, model, temperature, num_calls, use_web_search):
         if use_web_search:
             for main_content, sources in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature):
                 response = f"{main_content}\n\n{sources}"
-                logging.info(f"Generated Response (first line): {response.split('\n')[0]}")
+                first_line = response.split('\n')[0] if response else ''
+                logging.info(f"Generated Response (first line): {first_line}")
                 yield response
         else:
             for partial_response in get_response_from_pdf(message, model, num_calls=num_calls, temperature=temperature):
-                logging.info(f"Generated Response (first line): {partial_response.split('\n')[0]}")
+                first_line = partial_response.split('\n')[0] if partial_response else ''
+                logging.info(f"Generated Response (first line): {first_line}")
                 yield partial_response
     except Exception as e:
         logging.error(f"Error with {model}: {str(e)}")